D߾]!j l d Z ddlZddlZddlZddlZddlmZ ddlZddl m
Z
m Z m Z m
Z
mZmZmZ ddlmZ ddlmZ ej ej ej ej ej ej ej ej ej ej ej ej e!g
Z"g Z#e"D ]Z$dD ]Z% ej& e$ ' e% Z& ej( d ) e& Z*e#+ ej, g e& ej, d
e& e*e*- d e*- d j. e*- d ddd ddd
f g Ќg dZ/dd
gddgddggdfddgddgddggdfgZ0dddddg d fd d gfd d d!d"gfd#d$gZ1dd
gd%dd&d%d'gddgd(d
gfd)d*fd+d,ddgddggdfddgd-dd.d-d/gddgd
d(gfd0d1fd2d3ddgddggdfgZ2 ej, e0 ej& e/ ' d4 ej, e2 ej& e1 ' d4 ej, e0 ej& e/ ' d5 ej, e2 ej& e1 ' d5 ej3 d(d6g gZ4 G d7 d8e Z5d9 Z6d: Z7d; Z8d< Z9d= Z:d> Z;d? Z ? edAB ej> j@ dC ZAdD ZB ej& dEd*F ZC ej& dGdHgdIdIgd(dJgdK ZD ej& dLdMgdIeDgdN ZE ej& g dOdIgdz dN ZF ej& dGdHgdIdIgd(dJgdPdQgdR ZG ej& g g ddS ZHej> I dTeCeDeEeFeGeHg dU ZJej> K edVB dW ZLdX ZMdY ZNej> I dT ej& ej& dGej fdHej fdLej fgd*F dZ ej& d[ ej& dGdHgd\d\gddgdd]dZ d^fg ej& d_g ej& d[ ej& ej& ej& dGdHgd\d\gddgdd] dZf d^f fg ej& dG ej& ej& ej& ej& dGeOfdH ej& dGdHgd\d\gddgdd] fg dZf d^f d`f fg g da ZPdb ZQej> ? edAB dc ZRej> I dddedg df ZSdg ZTdh ZUg diZVg djZWdk ZXdl ZYdm ZZdn Z[do Z\dp Z]dq Z^dr Z_ds Z`ej> ? edtB ej> ? ej& eja jb dk duB ej> j@ edvw dx Zcdy Zddz Zed{ Zfej> I d| ej& dGdHgeg ej& d}d~di gdN d*f ej& eOd~di d1f ej& deOdffgd~di d1f ej& dGdHgeg ej& dLg ej& eOi gdN gdN d1fg ej> ? eoejh ji dk dB d ZjdS )aD 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': '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': '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': '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': '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': '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"
N)BytesIO)assert_assert_array_equal
assert_raisesassert_raises_regex assert_warnsIS_PYPYIS_WASM)requires_memory)formatz<>i dtype
) 2 )xi4r yf8r r zu1 g @g @ g @g @ r Info)valuec16)y2r Info2)nameS2)r$ r% r )y3r r )z3u4r r( )z2b1)colorr* info)NameU8)Valuer% r r @nny @ NNTcc)r8 r5 @ooy @OOFdd)r< r: <>)cz m+ & &~~3%%%%& &rO c t t z D ]1} | j t k r t | }t | | 2d S rP )rk rl r objectrc r rm s rN test_roundtrip_randsizerq sK m+ * * 9 %c**D sD ) ) )* *rO c p t D ]-} | j t k rt t t
| .d S rP )rk r rp r
ValueErrorre r] s rN test_roundtrip_truncatedru s@ @ @ 9 *&93 ? ? ?@ @rO c t j dt j t t j dz f } t
| }t | | d S )Nr7 r
)nponesr strr BUFFER_SIZEra r ) long_str_arr
long_str_arr2s rN
test_long_strr} sP 71BHc63E3I-J$K$KLLLL l++M|]33333rO z memmap doesn't work correctly)reasonc t t t z D ]\ }}|j j rt
j | d| d }t
j | d| d }t |d 5 }t j
|| d d d n # 1 swxY w Y |j j o |j j
}t j |d|j |j | }||d< | t |d 5 }| }d d d n # 1 swxY w Y t |d 5 }| } d d d n # 1 swxY w Y t% || t j |d
}| d S ) Nnormalz.npymemmapwbw+)moder shape
fortran_order.rbr)r ) enumeraterk rl r hasobjectospathjoinopenr rY flags f_contiguous c_contiguous open_memmapr flushrJ ri )
tmpdirir] nfnmfnfpr ma normal_bytes memmap_bytess
rN test_memmap_roundtripr s` L= 899 3 9 gll6#3A#3#3#344gll6#3A#3#3#344
#t__ ( r3 ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( (
I " A39+A'A
$ci&)i}N N N3
#t__ % 7799L % % % % % % % % % % % % % % %
#t__ % 7799L % % % % % % % % % % % % % % %lL111
#
.
.
.
5 s6 B**B. 1B. D;;D? D? E77E; >E; c : t j dd }t j | d }t j || t j | 5 }|d }d d d n # 1 swxY w Y t || d S )N zcompressed.npzrt r] ) rw rG randr r r savez_compressedloadr )r r] npz_filenpzarr1s rN test_compressed_roundtripr s
)..c
"
"Cw||F$455Hc**** c5z sD!!!!!s + B BBz
i1, i4, i1)alignabr )namesformatsoffsetsr@ d)r r )r r aabb)r r r titles)r r itemsizedtc X t j d| }t d D ]
}|dz ||< t j | d }t j || t j | 5 }|d }d d d n # 1 swxY w Y t || d S )Nr rC z aligned.npzrt r] ) rw zerosranger r r savezr r )r r r] r r r r s rN test_load_padded_dtyper s
(1b//C
1XX QAw||FM22HHX3 c5z sD!!!!!s : BBBz!Emscripten NODEFS has a buggy dupc b d} t j t j t d| }t j t d 5 t j | }d d d n # 1 swxY w Y t |t j d d S )Nzwin64python2.npydatazReading.*this warning\.matchr ) r r r dirname__file__pytestwarns UserWarningrw r r rx )fnamer r s rN %test_python2_python3_interoperabilityr s E
7<<1165 A AD k)C D D D wt}} tRWQZZ(((((s !BB Bc V t j t j t d } t j d t ddgt }dD ]>}t j | | }dD ]}t j |d| }|
d r |d
}| n|}|d k r| d rxt t |d
t t! |d d |d d t! |d | |d t t |d
t$ t! || | d r|
d rt j |d }t' t( |j d
| t j |ddd }t' t, |j d
| t' t( t
j |d t' t, t
j |ddd @d S )Nr u 優良s 不良r
)py2-objarr.npypy2-objarr.npzzpy3-objarr.npyzpy3-objarr.npz)byteslatin1TrW encodingz.npzr r py2r r rV F)rW fix_importsr )r r r r r rw arrayr rp r endswithclose
startswithr
isinstancery r encoder r UnicodeError __getitem__ ImportError)data_direxpectedr r r data_fr s rN test_pickle_python2_python3r sr w||BGOOH 5 5v>>Hxun46 $& & &H6 #1 #1w||He,,+ 3 3HWTxHHHF~~f%%
c{ 8##(8(8(?(?#
47C00111"49hssm<<<"48??8#<#>H
7<<"2 3 3D*bgt $x9 9 9 9 7<<"2 3 3D EH = = = 6j!-5556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 7<< 5 6 6D*bgtRXtfF-K-K-K $& & & & & &s = C%%C),C))r r i1)r r r r )r! )r rA r rB c t j | j }t || t j d| }t
| }t || d S )Nr )r descr_to_dtypedescrri rw r ra r )r dt1r r` s rN test_descr_to_dtyper X sW d
)
)C#r
8Ar??D T??DtT"""""rO c t } d t d D }t j d| }t j | |d t
j d 5 }t
j d d
t t j | | t |d j t u d d d n # 1 swxY w Y | d |
}t t | t j z d k | d t j | d
}t# || t% t& t j | |d d S )Nc , g | ]}d |z dz t fS z%dd float.0r s rN
z$test_version_2_0.. & 8 8 8D1H U
# 8 8 8rO r
r r versionTrecordalwaysr r @
max_header_sizer7 r )r r rw rx r rY warningscatch_warningsfilterwarningsr r categoryseekreadlinelen ARRAY_ALIGNr[ r r rs )r^ r r wheaderns rN test_version_2_0r s A 8 8U3ZZ 8 8 8B
B A
q!V,,,, - - - . "k:::1a !
,---. . . . . . . . . . . . . . . FF1III
ZZ\\F CKK&, , 1222FF1III!V444Aq! *f 0!Q?????s *ACC C c d t d D }t j d| }t j | d }t j | d }t
t t j |d|j
|j d t j |d|j
|j d
}||d < | t j |d d
}t || t j d 5 }t j ddt" t j |d|j
|j d }t% |d j t" u ||d < | d d d n # 1 swxY w Y t j |d d
}t || d S )Nc , g | ]}d |z dz t fS r r r s rN r z+test_version_2_0_memmap.. r rO r r r
zversion2_01.npyzversion2_02.npyr r )r r r r r .r r )r r Tr r r r )r rw rx r r r r rs r r r r r r r r r r r r )r r r tf1tf2r r s rN test_version_2_0_memmapr s
9 8U3ZZ 8 8 8B
B A
',,v 1
2
2C
',,v 1
2
2C *f 0#D"#'6; ; ; ;
Cd!'"#'6
;
;
;BBsGHHJJJ Cc6 B B BBr1 - - - "k:::
$ag&'gt= = =!
,---3
Cc6 B B BBr1 s A9F F #F mmap_moder c t j | d }t j dddz dz }t j t d 5 t j || d d d n # 1 swxY w Y t j t d 5 t j ||
d d d n # 1 swxY w Y t j t d 5 t j ||d d d d n # 1 swxY w Y t j ||d
}t || t j ||d }t || d S )Nzlarge_header.npyr7 i,' r r
.*format 2.0r
Header.*large)r
N )r
r T)r
rW )
r r r rw r r r r r raisesrs r r )r r
r^ r] ress rN test_huge_headerr s
V 011A
(1DJsN
+
+
+C k 8 8 8
3
z 9 9 9 ( (
Y''''( ( ( ( ( ( ( ( ( ( ( ( ( ( (
z 9 9 9 ? ?
Y>>>>? ? ? ? ? ? ? ? ? ? ? ? ? ? ?
'!yt
<
<
A> CC
C)D
DDc t j | d }t j dddz dz }t j t d 5 t j || d d d n # 1 swxY w Y t j t d
5 t j | d d d d n # 1 swxY w Y t j t d
5 t j |d
d d d d n # 1 swxY w Y t j |d d }t || t j |d
d }t || d S )Nzlarge_header.npzr7 r r
r r
r r rt r r] r r TrV r )
r r r rw r r r r r r rs r r )r r^ r] r s rN test_huge_header_npzr s
V 011A
(1DJsN
+
+
+C k 8 8 8
z 9 9 9
5
z 9 9 9 1 1
5)))%001 1 1 1 1 1 1 1 1 1 1 1 1 1 1
'!$
'
'
'
.CsC
'!V
,
,
,U
3CsC s6 A;;A?A?!CC C . DD Dc t } t j d }t j | |d t j | | t j | |d t j | | t j | |d t j | | g d}|D ]F}t t d 5 t j | || d d d n # 1 swxY w Y Gd S )Nr7 r r r ))r7 r7 )r r )r r7 r ) r z we only support format version.*)r rw aranger rY r rs )r^ r] bad_versionsr s rN test_write_versionr sJ A
)A,,C
q#v....
q#
q#t,,,,
q#
q#v....
q# L 8 8
!CE E 8 8 q#w 7 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 88 8s C((C, /C, )s NUMPYs NUMPY s NUMPY s NUMPY s NUMPYs NUMPY)s NUMPY s NUMPY s numpy s MATLB s NUMPYs NUMPYrO c | t } t }t j dt }t j | |d t j ||d | d | d t j | }t j | }t |dk t |dk t | t j
k t | t j
k d S )Nr r r
r r r r ) r rw rx r r rY r
read_magicr tell MAGIC_LEN)s1s2r] version1version2s rN test_read_magicr% s B B
'&
&
&
&C
r3////
r3////GGAJJJGGAJJJ $$H $$H H H BGGII) )*** BGGII) )*****rO c x t D ]1} t | }t t t j | 2d S rP )malformed_magicr r rs r r[ magicr^ s rN test_read_magic_bad_magicr* 2 s> 8 8 ENNj&"3Q77778 8rO c t t z D ]1} t | }t t t
j | 2d S rP )bad_version_magicr' r r rs r r[ r( s rN test_read_version_1_0_bad_magicr- 8 sC "_4 8 8 ENNj&"3Q77778 8rO c t t t j dd t t t j dd t t t j dd t t t j dd d S )Nr r7 )r rs r r) rO rN test_bad_magic_argsr1 > s` *flB222*flC333*flAr222*flAs33333rO c t } t ddd}t j | | t } ddz dfg|d< t t
t j | | d S )NFz O A
'&
&
&
&C
q#v....FF6 " 8 ; ;E7E AFFHHv) )Q ./// UGU #u'= =>>>>>rO c t } t j dt }t j | |d | t j t j | \ }}}t |
t j z dk t |||fddt fk d S )Nr r
r r r F) r rw rx r r rY r r read_array_header_2_0r r r r< s rN test_read_array_header_2_0rB \ r? rO c t } t t t j | t d } t t t j | t d } t t t j | t d } t t t j | ddddd}t } t j | | t t t j | d S ) N 1s s@ NUMPY 6 {'descr': 'x', 'shape': (1, 2), }
r7 r Fr r )r r r extrakey)r r rs r r; r6 r7 s rN test_bad_headerrG i s A*f :A>>>
A*f :A>>> A*f :A>>>
" A *f :A>>> A A
!!Q'''*f :A>>>>>rO c t j dk st j dk rt j d t j | d } dd l}| ddd|g n$# t $ r t j d Y nw xY wt |d
5 }| d t j
d }t j || d d d n # 1 swxY w Y t |d
5 }| d t j | }d d d n # 1 swxY w Y t! || d S )Nwin32cygwinz)Unknown if Windows has sparse filesystems sparse_filer truncatez-s
5368709120z Could not create 5GB large filer l rC r )sysplatformr skipr r r
subprocess
check_call Exceptionr r rw r r r r )r tf_namespr^ r r s rN test_large_file_supportrV s 3<8#;#; ?@@@gll6=11G7
z4 w?@@@@ 7 7 7 5666667
gt z IaLL
1
gt z GAJJ q! s6 A4 4 BB(?C33C7:C7*EE Ez
flaky on PyPyztest requires 64-bit systeml )
free_bytesc d} t j |t j }n$# t $ r t j d Y nw xY wt j | d }t |d 5 }t j
|| d d d n # 1 swxY w Y ~t |d 5 }t j | d }d d d n # 1 swxY w Y |j |k sJ d S ) N)i @r r
zCould not create large file
large_archiver rt r r] )
rw emptyuint8 MemoryErrorr rP r r r r r r r )r r r r r^ new_as rN test_large_archiver^ sz
E3 HU"( + + + 3 3 3 1222223
GLL 1 1E
eT a
eT "a
5!" " " " " " " " " " " " " " " ;% s3 % AA9B B #B 8C C#&C#c t j | d }t j | t j | 5 } d d d d S # 1 swxY w Y d S )Nz nothing.npz)r r r rw r r )r r npss rN test_empty_npzra s GLL / /EHUOOO
3
s AA A c t j g ddt fdt fg }t j | d }t
|d 5 }t j ||d d d d n # 1 swxY w Y t
|d 5 }t j | }d d d n # 1 swxY w Y t || t
|d 5 }t t 5 t j ||d d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S )
N)r7 r rE rc rE intu 整形r
z unicode.npyr )r r r r )
rw r rd r r r r r rY r[ r r r )r r] r r^ r` s rN test_unicode_field_namesre s>
(
GM C
GLL / /E
eT 3a1c622223 3 3 3 3 3 3 3 3 3 3 3 3 3 3
eT $a ##$ $ $ $ $ $ $ $ $ $ $ $ $ $ $sD!!!
eT 5a
+
&
& 5 5 q#t 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 55 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5sZ A==BBB99B= B=$D59D D5 D! !D5$D! %D55D9<D9c b g dg dg dg dfD ]\ } }}d t t j D D ]u}t }t j || rd|fn|df| t j d|z t fg d t | |k sJ vd S ) N)F )F )Tri rh )T rj c g | ]}d |z S )r r0 r s rN r z+test_header_growth_axis.. s IIIqRUIIIrO r r3 )
r r GROWTH_AXIS_MAX_DIGITSr r6 rw r rd r rZ )is_fortran_array dtype_spaceexpected_header_lengthrL r s rN test_header_growth_axisrr s ***OOO___B @ @=+'= JIE&*G$H$HIII @ @DB )"&6E!TT1I!1C OS#9":;;/ /
r{{}}%%)?????? @ @ @rO zdt, failS3somestuff)metadatasubarrayr ) r r z PyPy bug in error formattingc < t j d| }t }t t 5 t j || d d d n # 1 swxY w Y | d |rCt t 5 t j | d d d d S # 1 swxY w Y d S t j | }ddl
m } t || ||j
|j
usJ ||j
|j
u sJ d S )Nr r
r )
drop_metadata)rw rx r r r r r r rs r numpy.lib.utilsrz r r )r failr] bufr` rz s rN test_metadata_dtyper~ s
'"B
C
))C k " "
S HHQKKK 7
:
&
& GCLLL ws||1111113%%% }SY''sy8888 }TZ((DJ666666s# AA "A B33B7:B7)k__doc__rN r r r ior numpyrw
numpy.testingr r r r r r r
numpy.testing._private.utilsr numpy.libr r[ int8uint16int16uint32int32uint64int64float32float64 complex64
complex128rp scalarsrk scalarendianr newbyteorderr astypebasicextendr reshapeTPdescrPbufferTNdescrNbufferTr rl rE ra rc re ri rn rq ru r} markskipifslowr r r dt2dt3dt4dt5dt6 parametrizer xfailr r r rd r r r r r r r, r' r% r* r- r1 r9 r> rB rG rV intpr r^ ra re rr r implementationr r~ r0 rO rN r sW R Rf
9 8 8 8 8 8
HGIHIHIHJJLM
F --f55 $&&u-- BHRu % % % BHRu % % % MM( # # MM( # # % MM( # #DDbD##A#I .
4
Vr2hR
!1%Vr2hR
!1% ( !
#
. Vb"tb"XBx!Q8$
E :R2r(+Q0Vb"tb"XBx!Q8$
F :R2r(+Q0
BHXXRXf - - : :3 ? ?@@@ BHXXRXf - - : :3 ? ?@@@ BHXXRXf - - : :3 ? ?@@@ BHXXRXf - - : :3 ? ?@@@ BHQ12333
" " " " " " " " & & & * * *@ @ @ 4 4 4 G$CDD ED<" " " bh|4(((bh#st q6# # $ $ bh#ss <<==bh TF1H==>>bh#st q6dD\; ; < < bh::;;S#sC =>>" " ?>" 7#FGG) ) HG),1 ,1 ,1^
&
&
& BHXRXRW~RXRX $% % %
BHsHBHs3i)-d )*1)*
BH
BHsHBH BHs3i!%d !"1!"$ $
%
%
BH hbh BH BH c
hbhS 37+34Q%34(6 (6 7 7 8
;/ / /b# #c/ /b#@ @ @2 G$CDD ED < sDk22! ! 32!&! ! !(8 8 86 + + +*8 8 8 8 8 8 4 4 4C C C
?
?
?
?
?
?? ? ?: 2 GO44HBHRW%%.28 : : I&&& '& : : 54
,
5 5 5. @ @ @
RXc
!7+9- 9- 9- 1. / / 0 0157
RXcVW -...6
RX
S$K()VW4EFFFN
RXc
xrx3%XRXcB5O5O5O4PQQRR0
& GG 2 :i G- / /7 7/ / 7 7 7rO