Kc @ s d Z d d k Z d d k l Z d d k Z d d k Z d d k Z d d k Z d d k Td d k
l Z d a
d Z d Z e i e i e i e i e i e i e i e i e i e i e i e i e g
Z g Z x e D] Z x d D] Z e i! e i" e Z! e i# d i$ e! Z% e i& e i' g d
e! e i' d d
e! e% e% i( dR e% i( dS i) e% i( dT d d d d d d f g q Wq Wd d dU f d d dV f dW g Z* d d g d d g d d g g d f d d g d d g d d g g d f g Z+ d d dX f d dY dZ d d[ d d d\ f d# d d] f d$ d% d^ f g f d_ d` g f da d) db dc g f d d dd f de g Z, d d g d- d d. d- d/ g d d g d0 d g f d1 e- f d2 df d d g d d g g d f d d g d3 d d4 d3 d5 g d d g d d0 g f d6 e. f d7 dg d d g d d g g d f g Z/ e i' e+ d
e i! e* i" d8 e i' e/ d
e i! e, i" d8 e i' e+ d
e i! e* i" d9 e i' e/ d
e i! e, i" d9 g Z0 d: Z1 d; Z2 d< Z3 d= Z4 d> Z5 d? d@ dA dB dC dD g Z6 dE dF dG dH dI dJ dK g Z7 dL Z8 dM Z9 dN Z: dO Z; dP Z< e= dQ j o e> n d S(h sD Test the .npy file format.
Set up:
>>> from cStringIO import StringIO
>>> 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(StringIO(format.magic(1, 0)))
(1, 0)
>>> format.read_magic(StringIO(format.magic(0, 0)))
(0, 0)
>>> format.read_magic(StringIO(format.magic(255, 255)))
(255, 255)
>>> format.read_magic(StringIO(format.magic(2, 5)))
(2, 5)
Test the header writing.
>>> for arr in basic_arrays + record_arrays:
... f = StringIO()
... format.write_array_header_1_0(f, arr)
... 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': '|O4', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '|O4', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '|O4', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '|O4', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '|O4', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '|O4', 'fortran_order': False, 'shape': (3, 3)} \n"
"F\x00{'descr': '|O4', 'fortran_order': False, 'shape': (0,)} \n"
"F\x00{'descr': '|O4', 'fortran_order': False, 'shape': ()} \n"
"F\x00{'descr': '|O4', 'fortran_order': False, 'shape': (15,)} \n"
"F\x00{'descr': '|O4', 'fortran_order': False, 'shape': (3, 5)} \n"
"F\x00{'descr': '|O4', 'fortran_order': True, 'shape': (5, 3)} \n"
"F\x00{'descr': '|O4', '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"
iN( t StringIO( t *( t formatc C s t i a d S( N( t tempfilet mkdtempt tempdir( ( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt setup_module$ s c C s; t d j o* t i i t o t i t d a n d S( N( R t Nonet ost patht isdirt shutilt rmtree( ( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt teardown_module( s
s <>i t dtypei
i i i t xt i4t yt f8t zt u1g @g @i i g @g @i t Infot valuet c16t y2t Info2t namet S2t y3t z3t u4t z2t b1t colort infot Namet U8t Valuey @t nny @i t NNt ccy @t ooy @t OOt ddt c C s> t } t i | | t | i } t i | } | S( N( R R t write_arrayt getvaluet
read_array( t arrt ft f2t arr2( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt roundtrip s
c C s d S( N( ( t o1t o2( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt assert_equal s c c s3 x, t t D] } t | } t | | f Vq Wd S( N( t basic_arrayst
record_arraysR5 t assert_array_equal( R1 R4 ( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt test_roundtrip s c c sm t i d j p
t i d j pIxFt t D]6} | i i o q+ n t i i t d } t i i t d } t
| d } z t i | | Wd | i
X| i i o | i i } t i | d d d | i d | i d
| } | | d <~ t
| d } | i } | i
t
| d } | i } | i
t | | f Vt i | d d
} q+ Wn d S( Nt win32t cygwins
normal.npys
memmap.npyt wbt modes w+R t shapet
fortran_order.t rbt r( t syst platformR9 R: R t hasobjectR R t joinR t openR R. t closet flagst f_contiguoust c_contiguoust open_memmapRA t readR8 ( R1 t nfnt mfnt fpRB t mat normal_bytest memmap_bytes( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt test_memmap_roundtrip s0
c C s t } t i d } t i | | d d t i | | d d d
d d d
g } xQ | D]I } y t i | | d | Wn t j
o q] Xt d | f q] Wd S( Ni t versioni i i s9 we should have raised a ValueError for the bad version %r( i i ( i i ( i i ( i i ( i i ( i i ( i i ( R t npt arangeR R. t
ValueErrort AssertionError( R2 R1 t bad_versionsRW ( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt test_write_version_1_0 s" s NUMPYs NUMPY s NUMPY s NUMPY s NUMPYs NUMPYs NUMPY t NUMPY s numpy s MATLB s NUMPYs NUMPYt c c s; x4 t D], } t | } t t t i | f Vq Wd S( N( t malformed_magicR t raisesRZ R t
read_magic( t magicR2 ( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt test_read_magic_bad_magic s c c s? x8 t t D], } t | } t t t i | f Vq Wd S( N( t bad_version_magicR` R Ra RZ R R0 ( Rc R2 ( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt test_read_version_1_0_bad_magic s c C s\ t t t i d d t t t i d d t t t i d d t t t i d d d S( Nii i ( t
assert_raisesRZ R Rc ( ( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt test_bad_magic_args s c C ss t } h d d 6d d 6} t i | | t } h d d 6d d 6d d d d 6} t t t i | | d S( Ni t ai t bR i t c( R R t write_array_header_1_0Rg RZ ( t st d( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt test_large_header s #c C s t } t t t i | t d } t t t i | t d } t t t i | h d d 6d d 6} t } t i | | t t t i | h d d 6t d 6d d 6d d
6} t } t i | | t t t i | d S(
Nt 1s i i RA R t descrRB it extrakey( i i ( i i ( R Rg RZ R t read_array_header_1_0Rl t False( Rm Rn ( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt test_bad_header s$
t __main__( i i ( i i ( i i ( i ( i i ( R R ( i ( s valueR ( R R ( s nameR ( i ( i ( i ( s nameR ( R R ( R! R ( R# R$ ( R% R ( i i ( R R ( R' y @( R* y @(? t __doc__RE t cStringIOR R R R t numpyRX t
numpy.testingt numpy.libR R R R R
t uint8t int8t uint16t int16t uint32t int32t uint64t int64t float32t float64t complex64t
complex128t objectt scalarsR9 t scalart endianR t newbyteorderRY t astypet basict extendt arrayt reshapet Tt Pdescrt PbufferTt Ndescrt TrueRt t NbufferTR: R5 R8 R< RV R] Re R` Rd Rf Rh Ro Ru t __name__t run_module_suite( ( ( sA /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_format.pyt s
7 $* Z`$$$* %