%PDF-
%PDF-
Mini Shell
Mini Shell
a
z[yc˛ � @ sp d Z ddlZddlZddlZddlZddlmZ ddlZddl m
Z
m Z m Z m
Z
mZmZ ddlmZ ddlmZ ejejejejejejejejej ej ej ej e g
Z!g Z"e!D ]�Z#dD ]vZ$e�%e#��&e$�Z%e�'d��(e%�Z)e"�*ej+g e%d �ej+d
e%d �e)e)�,d �e)�,d �j-e)�,d �ddd �ddd
�f g� q�q�g d�Z.dd
gddgddggdfddgddgddggdfgZ/dddddg d �fd d gfd d d!d"gfd#d$gZ0dd
gd%dd&d%d'gddgd(d
gfd)d*fd+d,ddgddggdfddgd-dd.d-d/gddgd
d(gfd0d1fd2d3ddgddggdfgZ1ej+e/e�%e.��&d4�d �ej+e1e�%e0��&d4�d �ej+e/e�%e.��&d5�d �ej+e1e�%e0��&d5�d �ej2d(d6gd �gZ3G d7d8� d8e�Z4d9d:� Z5d;d<� Z6d=d>� Z7d?d@� Z8dAdB� Z9dCdD� Z:dEdF� Z;dGdH� Z<ej=j>dIdJ� �Z?dKdL� Z@ej%dMd*dN�ZAe�%dOdPgdQdQgd(dRgdS��ZBe�%dTdUgdQeBgdV��ZCe�%g dW�dQgd dV��ZDe�%dOdPgdQdQgd(dRgdXdYgdZ��ZEe�%g g dd[��ZFej=�Gd\eAeBeCeDeEeFg�d]d^� �ZHd_d`� ZIdadb� ZJdcdd� ZKej=�Gd\e�%ej%dOejfdPejfdTejfgd*dN�de�e�%dfe�%dOdPgdgdggddgddh�de�difg�e�%djg�e�%dfe�%e�%e�%dOdPgdgdggddgddh��def�dif�fg�e�%dOe�%e�%e�%e�%dOeLfdPe�%dOdPgdgdggddgddh��fg�def�dif�dkf�fg�g�dldm� �ZMdndo� ZNdpdq� ZOej=�Gdrdsdg�dtdu� �ZPdvdw� ZQdxdy� ZRg dz�ZSg d{�ZTd|d}� ZUd~d� ZVd�d�� ZWd�d�� ZXd�d�� ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]ej=j^e�%ej_�j`dk d�d��ej=j>ed�d��d�d�� ���Zad�d�� Zbd�d�� Zcej=�Gd�e�%dOdPgedej%d�d�d�id��gdV��d*fej%eLd�d�id��d1fej%d�eLd�ffgd�d�id��d1fe�%dOdPgede�%dTgej%eLi d��gdV��gdV��d1fg�ej=j^e�oZejejfd�kd�d��d�d�� ��ZgdS )�a�D Test the .npy file format.
Set up:
>>> import sys
>>> from io import BytesIO
>>> from numpy.lib import format
>>>
>>> scalars = [
... np.uint8,
... np.int8,
... np.uint16,
... np.int16,
... np.uint32,
... np.int32,
... np.uint64,
... np.int64,
... np.float32,
... np.float64,
... np.complex64,
... np.complex128,
... object,
... ]
>>>
>>> basic_arrays = []
>>>
>>> for scalar in scalars:
... for endian in '<>':
... dtype = np.dtype(scalar).newbyteorder(endian)
... basic = np.arange(15).astype(dtype)
... basic_arrays.extend([
... np.array([], dtype=dtype),
... np.array(10, dtype=dtype),
... basic,
... basic.reshape((3,5)),
... basic.reshape((3,5)).T,
... basic.reshape((3,5))[::-1,::2],
... ])
...
>>>
>>> Pdescr = [
... ('x', 'i4', (2,)),
... ('y', 'f8', (2, 2)),
... ('z', 'u1')]
>>>
>>>
>>> PbufferT = [
... ([3,2], [[6.,4.],[6.,4.]], 8),
... ([4,3], [[7.,5.],[7.,5.]], 9),
... ]
>>>
>>>
>>> Ndescr = [
... ('x', 'i4', (2,)),
... ('Info', [
... ('value', 'c16'),
... ('y2', 'f8'),
... ('Info2', [
... ('name', 'S2'),
... ('value', 'c16', (2,)),
... ('y3', 'f8', (2,)),
... ('z3', 'u4', (2,))]),
... ('name', 'S2'),
... ('z2', 'b1')]),
... ('color', 'S2'),
... ('info', [
... ('Name', 'U8'),
... ('Value', 'c16')]),
... ('y', 'f8', (2, 2)),
... ('z', 'u1')]
>>>
>>>
>>> NbufferT = [
... ([3,2], (6j, 6., ('nn', [6j,4j], [6.,4.], [1,2]), 'NN', True), 'cc', ('NN', 6j), [[6.,4.],[6.,4.]], 8),
... ([4,3], (7j, 7., ('oo', [7j,5j], [7.,5.], [2,1]), 'OO', False), 'dd', ('OO', 7j), [[7.,5.],[7.,5.]], 9),
... ]
>>>
>>>
>>> record_arrays = [
... np.array(PbufferT, dtype=np.dtype(Pdescr).newbyteorder('<')),
... np.array(NbufferT, dtype=np.dtype(Ndescr).newbyteorder('<')),
... np.array(PbufferT, dtype=np.dtype(Pdescr).newbyteorder('>')),
... np.array(NbufferT, dtype=np.dtype(Ndescr).newbyteorder('>')),
... ]
Test the magic string writing.
>>> format.magic(1, 0)
'\x93NUMPY\x01\x00'
>>> format.magic(0, 0)
'\x93NUMPY\x00\x00'
>>> format.magic(255, 255)
'\x93NUMPY\xff\xff'
>>> format.magic(2, 5)
'\x93NUMPY\x02\x05'
Test the magic string reading.
>>> format.read_magic(BytesIO(format.magic(1, 0)))
(1, 0)
>>> format.read_magic(BytesIO(format.magic(0, 0)))
(0, 0)
>>> format.read_magic(BytesIO(format.magic(255, 255)))
(255, 255)
>>> format.read_magic(BytesIO(format.magic(2, 5)))
(2, 5)
Test the header writing.
>>> for arr in basic_arrays + record_arrays:
... f = BytesIO()
... format.write_array_header_1_0(f, arr) # XXX: arr is not a dict, items gets called on it
... print(repr(f.getvalue()))
...
"F\x00{'descr': '|u1', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '|u1', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '|u1', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '|u1', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '|u1', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '|u1', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '|u1', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '|u1', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '|u1', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '|u1', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '|u1', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '|u1', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '|i1', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '|i1', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '|i1', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '|i1', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '|i1', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '|i1', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '|i1', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '|i1', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '|i1', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '|i1', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '|i1', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '|i1', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '<u2', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '<u2', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '<u2', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '<u2', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '<u2', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '<u2', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '>u2', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '>u2', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '>u2', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '>u2', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '>u2', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '>u2', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '<i2', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '<i2', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '<i2', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '<i2', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '<i2', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '<i2', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '>i2', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '>i2', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '>i2', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '>i2', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '>i2', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '>i2', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '<u4', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '<u4', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '<u4', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '<u4', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '<u4', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '<u4', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '>u4', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '>u4', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '>u4', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '>u4', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '>u4', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '>u4', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '<i4', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '<i4', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '<i4', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '<i4', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '<i4', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '<i4', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '>i4', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '>i4', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '>i4', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '>i4', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '>i4', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '>i4', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '<u8', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '<u8', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '<u8', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '<u8', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '<u8', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '<u8', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '>u8', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '>u8', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '>u8', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '>u8', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '>u8', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '>u8', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '<i8', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '<i8', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '<i8', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '<i8', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '<i8', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '<i8', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '>i8', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '>i8', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '>i8', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '>i8', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '>i8', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '>i8', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '<f4', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '<f4', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '<f4', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '<f4', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '<f4', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '<f4', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '>f4', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '>f4', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '>f4', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '>f4', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '>f4', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '>f4', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '<f8', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '<f8', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '<f8', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '<f8', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '<f8', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '<f8', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '>f8', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '>f8', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '>f8', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '>f8', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '>f8', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '>f8', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '<c8', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '<c8', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '<c8', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '<c8', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '<c8', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '<c8', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '>c8', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '>c8', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '>c8', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '>c8', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '>c8', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '>c8', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '<c16', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '<c16', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '<c16', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '<c16', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '<c16', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '<c16', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '>c16', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '>c16', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '>c16', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '>c16', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '>c16', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '>c16', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': 'O', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': 'O', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': 'O', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': 'O', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': 'O', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': 'O', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': 'O', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': 'O', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': 'O', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': 'O', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': 'O', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': 'O', 'fortran_order': False, 'shape': (3, 3)} \n"
"v\x00{'descr': [('x', '<i4', (2,)), ('y', '<f8', (2, 2)), ('z', '|u1')],\n 'fortran_order': False,\n 'shape': (2,)} \n"
"\x16\x02{'descr': [('x', '<i4', (2,)),\n ('Info',\n [('value', '<c16'),\n ('y2', '<f8'),\n ('Info2',\n [('name', '|S2'),\n ('value', '<c16', (2,)),\n ('y3', '<f8', (2,)),\n ('z3', '<u4', (2,))]),\n ('name', '|S2'),\n ('z2', '|b1')]),\n ('color', '|S2'),\n ('info', [('Name', '<U8'), ('Value', '<c16')]),\n ('y', '<f8', (2, 2)),\n ('z', '|u1')],\n 'fortran_order': False,\n 'shape': (2,)} \n"
"v\x00{'descr': [('x', '>i4', (2,)), ('y', '>f8', (2, 2)), ('z', '|u1')],\n 'fortran_order': False,\n 'shape': (2,)} \n"
"\x16\x02{'descr': [('x', '>i4', (2,)),\n ('Info',\n [('value', '>c16'),\n ('y2', '>f8'),\n ('Info2',\n [('name', '|S2'),\n ('value', '>c16', (2,)),\n ('y3', '>f8', (2,)),\n ('z3', '>u4', (2,))]),\n ('name', '|S2'),\n ('z2', '|b1')]),\n ('color', '|S2'),\n ('info', [('Name', '>U8'), ('Value', '>c16')]),\n ('y', '>f8', (2, 2)),\n ('z', '|u1')],\n 'fortran_order': False,\n 'shape': (2,)} \n"
� N)�BytesIO)�assert_�assert_array_equal�
assert_raises�assert_raises_regex� assert_warns�IS_PYPY)�requires_memory)�formatz<>i� ��dtype�
)� �2 ���� )��x�i4�r ��y�f8�r r ��zZu1� g @g @� � g @g @� r ZInfo)�value�c16)Zy2r ZInfo2)��name�S2)r r! r )Zy3r r )Zz3Zu4r r" )Zz2�b1)Zcolorr$ �info)�NameZU8)ZValuer! r r � @�nny @� �NNTZcc)r+ r( � @Zooy @�OOF�dd)r- r, �<�>)�c�z<f8�� r c s e Zd Zd� fdd� Z� ZS )�BytesIOSRandomSizeNc s dd l }|�d|�}t� �|�S )Nr r* )�randomZrandint�super�read)�self�sizer6 �� __class__� �A/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_format.pyr8 � s zBytesIOSRandomSize.read)N)�__name__�
__module__� __qualname__r8 �
__classcell__r= r= r; r> r5 � s r5 c C s0 t � }t�|| � t |�� �}tj|dd�}|S )NT�� allow_pickle�r r
� write_array�getvalue�
read_array��arr�f�f2�arr2r= r= r> � roundtrip� s
rN c C s, t � }t�|| � t|�� �}t�|�}|S �N)r r
rF r5 rG rH rI r= r= r> �roundtrip_randsize� s
rP c C s4 t � }t�|| � t |�� dd� �}t�|�}|S )Nr r rE rI r= r= r> �roundtrip_truncated� s
rQ c C s t | |k� d S rO )r )Zo1Zo2r= r= r> �
assert_equal_� s rR c C s$ t t D ]} t| �}t| |� qd S rO )� basic_arrays�
record_arraysrN r �rJ rM r= r= r> �test_roundtrip� s rV c C s. t t D ] } | jtkrt| �}t| |� qd S rO )rS rT r �objectrP r rU r= r= r> �test_roundtrip_randsize� s
rX c C s$ t D ]} | jtkrttt| � qd S rO )rS r rW r �
ValueErrorrQ �rJ r= r= r> �test_roundtrip_truncated� s
r[ c C s4 t jdt �ttjd f�d�} t| �}t| |� d S )Nr* r )�np�onesr �strr
� BUFFER_SIZErN r )Z long_str_arrZ
long_str_arr2r= r= r> �
test_long_str� s r` c
C sH t tt �D �]4\}}|jjr q tj�| d|� d��}tj�| d|� d��}t|d�� }t �
||� W d � n1 sx0 Y |j j o�|j j
}t j|d|j|j|d�}||d< |�� t|d��}|�� }W d � n1 s�0 Y t|d��}|�� } W d � n1 �s0 Y t|| � t j|d d
�}|�� q d S ) NZnormalz.npyZmemmap�wb�w+)�moder �shape�
fortran_order.�rb�r)rc )� enumeraterS rT r Z hasobject�os�path�join�openr
rF �flags� f_contiguous� c_contiguous� open_memmaprd �flushr8 rR )
�tmpdir�irJ ZnfnZmfn�fpre �maZ normal_bytesZ memmap_bytesr= r= r> �test_memmap_roundtrip� s* *� � & (
rv c C sj t j�dd�}tj�| d�}t j||d� t �|��}|d }W d � n1 sR0 Y t||� d S )N�� zcompressed.npzrZ rJ ) r\ r6 Zrandri rj rk Zsavez_compressed�loadr )rr rJ �npz_file�npz�arr1r= r= r> �test_compressed_roundtrip� s &r| z
i1, i4, i1)Zalign�a�br � )�names�formats�offsetsr1 �d)r� r� )r} � r~ ZaaZbb)r� r� r� Ztitles)r� r� �itemsize�dtc C s� t �d|�}td�D ]}|d ||< qtj�| d�}t j||d� t �|��}|d }W d � n1 sj0 Y t||� d S )Nr r4 z aligned.npzrZ rJ ) r\ �zeros�rangeri rj rk �savezrx r )rr r� rJ rs ry rz r{ r= r= r> �test_load_padded_dtype s &r� c C s: d} t j�t j�t�d| �}t�|�}t|t�d�� d S )Nzwin64python2.npy�datar ) ri rj rk �dirname�__file__r\ rx r r] )�fnamerj r� r= r= r> �%test_python2_python3_interoperability s
r� c C s� t j�t j�t�d�} tjd tddgtd�}dD �]L}t j�| |�}dD ]�}tj |d|d�}|�
d �rx|d
}|� � n|}|d kr�|� d �r�t
t|d
t�� t|d d� |d d� � t|d �|�|d � qHt
t|d
t�� t||� qH|� d �r0|�
d ��rVtj |dd�}tt|jd
� |� � tj |ddd d�}tt|jd
� |� � q0tttj |dd� tttj |ddd d� q0d S )Nr� u 優良s 不良r )�py2-objarr.npy�py2-objarr.npzzpy3-objarr.npyzpy3-objarr.npz)�bytes�latin1T�rD �encodingz.npzr r� Zpy2r r rC F)rD Z fix_importsr� )ri rj rk r� r� r\ �arrayr� rW rx �endswith�close�
startswithr �
isinstancer^ r �encoder� r � UnicodeError� __getitem__� ImportError)�data_dirZexpectedr� rj r� Zdata_fr� r= r= r> �test_pickle_python2_python3 sJ
��
�
�
�r� c C s� t j�t j�t�d�}t j�|d�}tttj|ddd� t j�|d�}tj|ddd�� }tt|j d� W d � n1 sz0 Y t j�| d�}tttj
|tj d gt d �dd
� d S ) Nr� r� Fr� r� r� r zpickle-disabled.npyr rC )
ri rj rk r� r� r rY r\ rx r� �saver� rW )rr r� rj rK r= r= r> �test_pickle_disallowF s
�,�r� )r r Zi1)r� r� r� r� )r )r r2 r r3 c C s8 t �| j�}t|| � t�d| �}t|�}t||� d S )Nr )r
Zdescr_to_dtype�descrrR r\ r� rN r )r� �dt1r{ rM r= r= r> �test_descr_to_dtypeU s
2
r� c C s� t � } dd� td�D �}tjd|d�}tj| |dd� tjdd ��<}t�d
d t � t�| |� t
|d j t u � W d � n1 s�0 Y | � d � | �
� }t
t|�tj d k� | � d � tj| d
d�}t||� tttj| |d� d S )Nc S s g | ]}d | d t f�qS �z%d�d ��float��.0rs r= r= r> �
<listcomp>� � z$test_version_2_0.<locals>.<listcomp>�� �� r �r r ��versionT��record�alwaysr� r �@
��max_header_size�r* r )r r� r\ r] r
rF �warnings�catch_warnings�filterwarnings� UserWarningr �category�seek�readline�len� ARRAY_ALIGNrH r r rY )rK r� r� �w�header�nr= r= r> �test_version_2_0� s 0
r� c C s. dd� t d�D �}tjd|d�}tj�| d�}tj�| d�}tttj |d|j
|j d d
� tj |d|j
|j d d
�}||d < |� � tj |d
dd�}t
||� tjdd��X}t�ddt� tj |d|j
|j d d
�}t|d jtu � ||d < |� � W d � n1 �s0 Y tj |d
dd�}t
||� d S )Nc S s g | ]}d | d t f�qS r� r� r� r= r= r> r� � r� z+test_version_2_0_memmap.<locals>.<listcomp>r� r� r zversion2_01.npyzversion2_02.npyrb r� )rc r rd r� r� .rg r� )rc r� Tr� r� r� r )r� r\ r] ri rj rk r rY r
rp r rd rq r r� r� r� r� r r� )rr r� r� Ztf1Ztf2ru r� r= r= r> �test_version_2_0_memmap� s0 � �
�(r� � mmap_moderg c C s t j�| d�}tjddd d d�}tjtdd�� t�||� W d � n1 sT0 Y tj t
d d�� tj ||d
� W d � n1 s�0 Y tj t
d d�� tj ||d d � W d � n1 s�0 Y tj ||d
d�}t ||� tj ||dd �}t ||� d S )Nzlarge_header.npyr* �i,�' rs r � .*format 2.0��match�
Header.*large)r� � N )r� r� T)r� rD � � )
ri rj rk r\ r� �pytest�warnsr� r� �raisesrY rx r )rr r� rK rJ �resr= r= r> �test_huge_header� s *,.
r� c C s t j�| d�}tjddd d d�}tjtdd�� tj||d � W d � n1 sV0 Y tj t
d
d�� t� |�d W d � n1 s�0 Y tj t
d
d��" tj |d d
�d W d � n1 s�0 Y tj |dd�d }t ||� tj |dd
�d }t ||� d S )Nzlarge_header.npzr* r� r� rs r r� r� rZ r� rJ r� r� TrC r� )
ri rj rk r\ r� r� r� r� r� r� rY rx r )rr rK rJ r� r= r= r> �test_huge_header_npz� s ,,0
r� c C s� t � } t�d�}tj| |dd� t�| |� tj| |d d� t�| |� tj| |dd� t�| |� g d�}|D ]>}ttd�� tj| ||d� W d � qp1 s�0 Y qpd S )Nr* r� r� r� ))r* r* )r r )r r* r )� r� z we only support format version.*)r r\ �aranger
rF r rY )rK rJ Z bad_versionsr� r= r= r> �test_write_version� s
�r� )s �NUMPYs �NUMPY s �NUMPY s �NUMPY s �NUMPYs �NUMPY��)s �NUMPY s NUMPY s �numpy s �MATLB s �NUMPYs �NUMPYr� c C s� t � } t � }tjdtd�}tj| |dd� tj||dd� | �d� |�d� t�| �}t�|�}t|dk� t|dk� t| � � tj
k� t|� � tj
k� d S )N�r r r r� r� r� r ) r r\ r] r� r
rF r� Z
read_magicr �tell� MAGIC_LEN)�s1�s2rJ Zversion1Zversion2r= r= r> �test_read_magic s
r� c C s$ t D ]} t| �}tttj|� qd S rO )�malformed_magicr r rY r
rH ��magicrK r= r= r> �test_read_magic_bad_magic. s r� c C s( t t D ]} t| �}tttj|� qd S rO )�bad_version_magicr� r r rY r
rH r� r= r= r> � test_read_version_1_0_bad_magic4 s r� c C sD t ttjdd� t ttjdd� t ttjdd� t ttjdd� d S )Nr r* � )r rY r
r� r= r= r= r> �test_bad_magic_args: s r� c C sF t � } ddd�}t�| |� t � } dddd d�}tttj| |� d S )Nr* r )r} r~ A xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr� )r} r~ r1 )r r
�write_array_header_1_0r rY ��sr� r= r= r> �test_large_headerA s
r� c C sr t � } tjdtd�}tj| |dd� | �tj� t�| �\}}}t | �
� tj dk� t |||fddtfk� d S )Nr� r r� r� r F) r r\ r] r� r
rF r� r� �read_array_header_1_0r r� r� �r� rJ rd Zfortranr r= r= r> �test_read_array_header_1_0K s r� c C sr t � } tjdtd�}tj| |dd� | �tj� t�| �\}}}t | �
� tj dk� t |||fddtfk� d S )Nr� r r� r� r F) r r\ r] r� r
rF r� r� Zread_array_header_2_0r r� r� r� r= r= r> �test_read_array_header_2_0X s r� c C s� t � } tttj| � t d�} tttj| � t d�} tttj| � ddd�}t � } t�| |� tttj| � ddddd�}t � } t�| |� tttj| � d S ) N� 1s �r* r r )rd r� Fr )rd re r� Zextrakey)r r rY r
r� r� r� r= r= r> �test_bad_headere s( � � r� c C s� t jdkst jdkr t�d� tj�| d�}z dd l}|�ddd|g� W n t yf t�d � Y n0 t
|d
��0}|� d � t �
d �}t �||� W d � n1 s�0 Y t
|d
��$}|� d � t �|�}W d � n1 s�0 Y t||� d S )N�win32�cygwinz)Unknown if Windows has sparse filesystemsZ sparse_filer �truncatez-sZ
5368709120z Could not create 5GB large filera l r4 rf )�sys�platformr� �skipri rj rk �
subprocessZ
check_call� Exceptionrl r� r\ r� r� rx r )rr Ztf_name�sprK r� rg r= r= r> �test_large_file_support� s
*
(r� ztest requires 64-bit system)�reasonl )Z
free_bytesc C s� d}zt j|t jd�}W n ty4 t�d� Y n0 tj�| d�}t |d�� }t j
||d� W d � n1 sr0 Y ~t |d�� }t � |�d }W d � n1 s�0 Y d S ) N)i @r r zCould not create large fileZ
large_archivera rZ rf rJ ) r\ �empty�uint8� MemoryErrorr� r� ri rj rk rl r� rx )rr rd r} r� rK Znew_ar= r= r> �test_large_archive� s , ,r c C sF t j�| d�}t�|� t�|��}W d � n1 s80 Y d S )Nz nothing.npz)ri rj rk r\ r� rx )rr r� Znpsr= r= r> �test_empty_npz� s
r c C s
t jg d�dtfdtfgd�}tj�| d�}t|d�� }tj||dd� W d � n1 s\0 Y t|d ��}t� |�}W d � n1 s�0 Y t
||� t|d��H}t t �� tj||d d� W d � n1 s�0 Y W d � n1 s�0 Y d S )
N)�r* r r� r r� �intu 整形r z unicode.npyra )r r r� rf )
r\ r� r ri rj rk rl r
rF rH r r r� )rr rJ r� rK rM r= r= r> �test_unicode_field_names� s
�� . (
r zdt, failZS3Zsome�stuff)�metadataZsubarrayr )� r r z PyPy bug in error formattingc C s� t jd| d�}t� }tt�� t �||� W d � n1 s>0 Y |�d� |r�tt�� t � |� W d � q�1 s~0 Y n t � |�}ddl
m } t ||� d S )Nr
r r )�
_has_metadata)
r\ r] r r r� r� r� r rY rx Znumpy.lib.formatr r )r� ZfailrJ �bufrM r r= r= r> �test_metadata_dtype� s
*
*
r )h�__doc__r� ri r� r� �ior Znumpyr\ Z
numpy.testingr r r r r r Z numpy.testing._private.utilsr Z numpy.libr
r� Zint8Zuint16Zint16Zuint32Zint32Zuint64Zint64Zfloat32Zfloat64Z complex64Z
complex128rW ZscalarsrS ZscalarZendianr Z newbyteorderr� ZastypeZbasic�extendr� Zreshape�TZPdescrZPbufferTZNdescrZNbufferTr� rT r5 rN rP rQ rR rV rX r[ r` ZmarkZslowrv r| r� Zdt2Zdt3Zdt4Zdt5Zdt6Z parametrizer� r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� ZskipifZintpr� r r r r� �implementationr� r r= r= r= r> �<module> s� �
�
�
�
��&�&�� �
� �
/������
���
���
�
���1
�� �
���
�
Zerion Mini Shell 1.0