U
|
O ±d5Q ã @ s¦ d dl Zd dlZd dlmZ d dlmZ d dlmZ d dl Z d dl
|
Z d dl m Z mZmZmZmZmZ d dlmZ G dd dZG d d
|
d
|
ZG d d d Zd d ZdS )é N)ÚBytesIO)Úpath)ÚPath)Úassert_Ú assert_equalÚassert_array_equalÚassert_array_almost_equalÚ assert_raisesÚtemppath)Úpicklec @ s¤ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'S )(ÚTestFromrecordsc C s~ t jjdddgdddggdd}t|d ¡ d
|
t|d jjd t|d jjd t|d jjd t|d jjd d S )NéÈ Údbeç333333ó?é ÚdeçÍÌÌÌÌÌô?zcol1,col2,col3©Únamesr )r r r Úcol1ÚiÚcol2ÚUé Úcol3Úf)ÚnpÚrecÚ fromrecordsr ÚitemÚdtypeÚkindÚitemsize©ÚselfÚr© r& úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_records.pyÚtest_fromrecords s ÿz TestFromrecords.test_fromrecordsc C s0 dt fdt fg}tjjg |d}t|jd dS )z0 Verify fromrecords works with a 0-length input ÚaÚb©r ©r N)Úfloatr r r r Úshape)r$ r r% r& r& r' Útest_fromrecords_0len s z%TestFromrecords.test_fromrecords_0lenc C s² dddgdddgg}ddd gd
|
d d gg}d d d
|
gd ddgg}t jj|d tfdtfgd}t|d | t|d | t jj|d dgd}t|d | t|d | t|| d S )N)é r )é é )é é )r4 r3 )r2 r1 )r r0 r0 r1 r3 r4 r2 r r) r* r+ r )r r r Úintr )r$ ÚdataZ
|
expected_aZ
|
expected_bÚr1Úr2r& r& r' Útest_fromrecords_2d s þ z#TestFromrecords.test_fromrecords_2dc C s* t jjddddd}t|d ¡ d d S )Ns¼ abcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgzi2,a3,i4r1 Úbig©Úformatsr. Ú byteorderr0 )idc s efgidcba©r r Úarrayr r r# r& r& r' Útest_method_array4 s z!TestFromrecords.test_method_arrayc C s4 t jjdddddddgdd }t|d
|
¡ d d S ) N©r0 é r) ©r é r* ©r1 é! Úc©r2 é, Úd©r3 é7 Úex©r4 éB r ©é éM Úgúu1,f4,a1©r< r0 )r g 6@ó br> r# r& r& r' Útest_method_array28 s ÿÿz"TestFromrecords.test_method_array2c C s> t jjdddddddgdd }t|d
|
d d
d
|
¡ d d S ) NrA rC rE rH rK rN rP rT rU r0 r )r2 g F@ó dr> r# r& r& r' Útest_recarray_slices= s ÿÿz$TestFromrecords.test_recarray_slicesc C s t ddddg¡}t ddddg¡}t d dddg¡}t jj|||gd
|
d }t|d ¡ d d |d< t|jt ddddg¡ d S )Nr0 r r1 r2 r) ÚddÚxyzZ12çñ?za,b,cr )r rZ ç @é" )r r? r Ú
|
fromarraysr r r) )r$ Úx1Zx2Zx3r% r& r& r' Útest_recarray_fromarraysB s z(TestFromrecords.test_recarray_fromarraysc C sÀ t t t¡d¡}t |d¡}t|d}| d¡ tjj|dddd}| d¡ tjj |dddd}| d¡ t
|
}| | ¡ ¡ | d ¡ tjj|dddd}| ¡ t|| t|| d S )
|
Nr6 zrecarray_from_file.fitsÚrbi úf8,i4,a5r1 r: r; r )r ÚjoinÚdirnameÚ__file__ÚopenÚseekr r Úfromfiler? r ÚwriteÚreadÚcloser )r$ Údata_dirÚfilenameÚfdr7 r8 Z bytes_arrayZr3r& r& r' Útest_recarray_fromfileK s
|
|
|
|
|
|
z&TestFromrecords.test_recarray_fromfilec C s¾ d}t j|dd}t j|dd}t j|dd}tt|D ]}ttdd||< q:t jj|||gdd}tt|D ]D}t|j| ttddk t|j | dk t|j
|
| dk qtd S ) Né
|
ÚOr+ Úf8r0 zdate,data1,data2r g ) r ÚzerosÚrangeÚlenÚlistr r_ r ÚdateZdata1Zdata2)r$ Úcountr) r* rG r Zminer& r& r' Útest_recarray_from_obj\ s z&TestFromrecords.test_recarray_from_objc C s¾ t jddgddgd}t j |¡}tt|t d¡ t t jddd¡}ttt j |¡ d ¡ t j t jd
|
d d¡}tt|
|
d ¡d t j t jd
|
dd¡}tt|
|
d¡d k d S )N)r0 g¹?)r gÉ?)Úfooú<i4)Úbarz<f8r+ zi rec.array([(1, 0.1), (2, 0.2)],
|
dtype=[('foo', '<i4'), ('bar', '<f8')])r2 rs z rec.arrayr1 úi4,i4z numpy.recordéÿÿÿÿÚi4z dtype=int32) r r? r r ÚreprÚtextwrapÚdedentÚonesr Ú
|
startswithÚfind©r$ r) r& r& r' Útest_recarray_reprj s
|
ÿ þz"TestFromrecords.test_recarray_reprc C sl t jjddd}tt|t d¡ |d }tt|d z t jd d tt|d
|
W 5 t jdd X d S ) N)r0 r] Z2003z<i4,<f8,<M8[Y]r+ zt rec.array((1, 2., '2003'),
|
dtype=[('f0', '<i4'), ('f1', '<f8'), ('f2', '<M8[Y]')])r& z(1, 2., '2003')F)Úlegacyz1.13z#(1, 2.0, datetime.date(2003, 1, 1)))r r r? r r r r Zset_printoptions)r$ Zarr_0dÚrecordr& r& r' Útest_0d_recarray_repr s z%TestFromrecords.test_0d_recarray_reprc C s t jddgdtfdgd}t j |¡}| t j¡}| t t j|jf¡¡}tdt | dt i}tdt | dt i}tdt | dt i}t
|
t |t j t
|
|jj t j t
|
|| t
|
t |t j t
|
|jj t j t
|
|| t
|
t |t j t
|
|jj t j t
|
|| d S )N©r0 ÚABC©r ZDEFr{ ©r} ÚS4r+ znumpy.Únumpy) r r? r5 r ÚviewÚrecarrayr r Úevalr r ÚtypeÚndarray)r$ r) Z recordarrZrecarrZ
|
recordviewZ recordarr_rZrecarr_rZ recordview_rr& r& r' Útest_recarray_from_repr s$
|
|
ÿ
|
|
z'TestFromrecords.test_recarray_from_reprc C s t jddgdtfdgd}t jdddd d
|
gt jd}tt j |¡jjt j ttt j |¡t j tt j |¡jjt j ttt j |¡t j t|
|
t j ¡jjt j tt|
|
t j ¡t j t|
|
t j ¡jjt j tt|
|
t j ¡t j t j t j d d d¡}|
|
d ¡
|
d ¡}tt|t j t|jjt j t j t j d d ddgd¡}t|d jjt j tt|d t j G dd dt j }|
|
|¡}tt|d | ddddg}t jjddddgdfddddgdfg|d}t|j jjt j t|j jjt j t|jjjt j t|jjjt j t j t j d dd¡}t|
|
d¡jjt j t|
|
d¡jjt j t|
|
d ¡jjt j t|
|
d ¡jjt j t j d d dt j d d dg}|D ]P} t j | ¡}
|
|
|
|
|
|
jjpÜ|
|
jt j¡} t| jj| jj tt| t| qºd S ) Nr r r{ r r+ r0 r r1 r2 r3 zf4,i4rs )r) r )r* r )rG r~ rG c @ s e Zd ZdS )z.TestFromrecords.test_recarray_views.<locals>.CN)Ú__name__Ú
|
__module__Ú __qualname__r& r& r& r' ÚCÃ s r )r) úf4,f4)r* ÚV8)rG )Zf4r )rJ ©Úi8r~ ©r0 r0 s 11111111r~ r )r r r r )r r? r5 Úint64r r r r r r r r r) r* ZvoidrG Úfloat32rJ Úint32Úfieldsr ) r$ r) r* r% Úrvr rG Z
|
test_dtypeZarrsÚarrr Zarr2r& r& r' Útest_recarray_views¥ s^
|
|
ÿÿ
|
|
ÿÿÿ z#TestFromrecords.test_recarray_viewsc C s~ t jjdddgdd}t jjdddgdd}t|j|jk t|j|jk tt|D ] }t|| ¡ || ¡ k qXd S )N)r0 Úabcg À @r )r Zxyg ÀÌÌ@r0 )r ú g Ù?r zc1, c2, c3, c4r )
|
r r r? r r r r. ru rv r )r$ ÚraÚpaÚkr& r& r' Útest_recarray_from_namesã s$ ýüýüz(TestFromrecords.test_recarray_from_namesc C s¸ t jjdddgdd}dddg|_t|d dddg tt|jt|jk d
|
|_t|jd
|
k d d d g|_t |d d d d gg d|_
|
t |d dddgg tt |j
|
t j j d S )N)r0 r¨ gffffff@)r r[ gÍÌÌÌÌÌ@)r1 Zwrsr zfield, shape, meanr r\ g@gffffff
|
@Úmean©r0 r1 ÚAÚBr r. r3 Úfield)r r r? r® r r r Úvarr. r r² Ú
|
isinstanceÚ collectionsr¨ ÚCallable)r$ rª r& r& r' Ú test_recarray_conflict_fieldsó s
|
ÿþ z-TestFromrecords.test_recarray_conflict_fieldsc C sÎ t jjddgdtfdtfgd}t|jddg t|d jd t|jdd g t|d
|
jd t dtfdtfg¡}t jjddg|d}t|jddg t|d jd t|jdd g t|d
|
jd d S ) N)r0 r) )r Úbbbr) r* r+ r0 r r r¸ r ) r r r r5 Úobjectr r) r* r )r$ r) Zndtyper& r& r' Ú$test_fromrecords_with_explicit_dtype s ÿz4TestFromrecords.test_fromrecords_with_explicit_dtypec C sD t jddgddtfgd}| t j¡}t|jd |jd kd d S ) N)úabc r0 )r¨ r ©r{ r r} r+ r r0 F)r r? r5 r r r r{ r r& r& r' Útest_recarray_stringtypes s
|
|
|
ÿ z)TestFromrecords.test_recarray_stringtypesc C sâ t d¡dft d¡dfd}t jjddgddd tfd
|
tfgfd tfd |fgd }tt|jt j tt|d t j tt|j t j
|
tt|d t j
|
t|j jjt j tt|d t j
|
t|j jjt j tt |j jj| tt|jt j tt|d t j tt|d j t j tt|d d t j t|d j jd t|d j d d t|d d jd t|d d d d t|d j jd t|d j d d t|d d jd t|d d d d d S )NZS5r r4 )r ÚD)r» r r0 )ÚabcdeZfgehi)r¨ )r r1 r0 )r¿ Zjklmnr¼ r} r° r± ÚbazÚquxr+ r{ r0 s fgehir¾ )r r r r? r5 r r r{ r r} r r rÁ Údictr¤ rÀ r° r¾ )r$ Z
|
qux_fieldsr) r& r& r' Útest_recarray_returntypes s< ÿ þþz)TestFromrecords.test_recarray_returntypesc C s dgd dgd g}t j |¡}t|d dddg t|d dddg t ddg¡}t jj||d}t|jd t|d d
|
d
|
d
|
g t|d d d d g d S ) NÚtestr1 Ú Úf0Úf1)rÆ z|S4)rÇ z|Sr+ r2 s testó )r r r_ r r r" )r$ Úcolsr Údtr& r& r' Útest_zero_width_strings4 s z'TestFromrecords.test_zero_width_stringsN)r r r r( r/ r9 r@ rW rY ra rp rz r r r r§ r r· rº r½ rà rË r& r& r& r' r s& > r c @ s e Zd Zdd ZdS )Ú TestPathUsagec
|
C s t ddr}t|}tj d¡ tj d¡ d¡}d|d< | d¡}| |¡ W 5 Q R X tj j
|
j |ddd }t || W 5 Q R X d S )
|
Nz.bin)Úsuffixé{ rq rc )g à?rq r¿ r3 Úwb)r< r. ) r
|
r r ÚrandomÚseedZrandZastyperg ZtofileÚcoreÚrecordsri r )r$ r r) ro Úxr& r& r' Útest_tofile_fromfileE s
|
þz"TestPathUsage.test_tofile_fromfileN)r r r rÕ r& r& r& r' rÌ C s rÌ c @ s¼ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zej d d d d g¡d!d" Zej d d d d g¡d#d$ Zd%d& Zd'S )(Ú
|
TestRecordc C s t jjddgdddgd| _d S )N)r0 r r1 )r2 r3 r4 ©r r| ©r r| )r r| r+ )r r r r6 ©r$ r& r& r' Ú setup_methodT s þÿzTestRecord.setup_methodc C s4 | j }t|jd d d|d _t|jd d d S ©Nr r0 ©r6 r r r r& r& r' Útest_assignment1Z s
|
zTestRecord.test_assignment1c C s4 | j }t|jd d d|jd< t|jd d d S rÛ rÜ r r& r& r' Útest_assignment2` s
|
zTestRecord.test_assignment2c C s | j }dd }tt|| d S )Nc S s d| d _ d S )Nr0 r )Zcol5)rÔ r& r& r' Úassign_invalid_columni s zATestRecord.test_invalid_assignment.<locals>.assign_invalid_column)r6 r ÚAttributeError)r$ r) rß r& r& r' Útest_invalid_assignmentf s z"TestRecord.test_invalid_assignmentc C sp t jjddgdgd}d|j_tt ddg|_W 5 Q R X tt" |jddgf|j j
|
d W 5 Q R X d S ) Nr, )r0 )r r r+ Fr r1 r ) r r r? ÚflagsÚ writeabler Ú
|
ValueErrorr Zsetfieldr r¤ r# r& r& r' Útest_nonwriteable_setfieldn s
|
|
z%TestRecord.test_nonwriteable_setfieldc sb | j ddg }t|jjd t|jjdddg | j ddg t jjd tt fdd d S )
|
Nr r )r r r× rØ )rÅ z|V4)r r c s j jS )N)r Údescrr& ©Úyr& r' Ú<lambda> rÈ z5TestRecord.test_out_of_order_fields.<locals>.<lambda>)r6 r r r ræ r rä )r$ rÔ r& rç r' Útest_out_of_order_fieldsw s ÿz#TestRecord.test_out_of_order_fieldsc C s| t jdg fgdt jfdt jdfgd}tdtjd D ]@}t|t tj||d¡ t|d t tj|d |d¡ q6d S )Nr0 r) r* r r+ r ©Úprotocol) r r? r£ ru r ÚHIGHEST_PROTOCOLr ÚloadsÚdumps©r$ r) Úprotor& r& r' Ú test_pickle_1 s &ÿzTestRecord.test_pickle_1c C s\ | j }tdtjd D ]@}t|t tj||d¡ t|d t tj|d |d¡ qd S )Nr r0 rë r )r6 ru r rí r rî rï rð r& r& r' Ú test_pickle_2 s ÿzTestRecord.test_pickle_2c C sh | j }tdtjd D ]L}t tj|d |d¡}t|jj t|jj t|jj
|
t|jj qd S )Nr r0 r rë ) r6 ru r rí rî rï r râ Ú c_contiguousÚ f_contiguousrã Zaligned)r$ r) rñ r« r& r& r' Ú test_pickle_3 s zTestRecord.test_pickle_3c
|
C sÖ t ddg¡}t jd|d}tdf}||d< d|d< |d ¡ \}}|t jjjksXt|\}}t |t
|
rntt t ||d
|
t |d ¡}t |¡} |d | ks¤tt t¡ |t d ¡||ksÈtW 5 Q R X d S ) N)Úobjrr )r5 r r0 r+ s emanr÷ é* r5 r é rr )r r ÚemptyÚ bytearrayÚ
|
__reduce__rÒ Z
|
multiarrayZscalarÚAssertionErrorr´ Úbytesr Ú RuntimeErrorr rï rî ÚpytestZwarnsÚDeprecationWarning)
|
r$ rÊ r) r6 ZctorÚargsr r÷ ÚdumpZ unpickledr& r& r' Útest_pickle_void s
|
|
zTestRecord.test_pickle_voidc C sf t ddg¡}t jd|d t j¡}t dddg¡|_t jdd tfd
|
tfd t fgd}|d d
|
g d S ) N)r{ r )r} rr r¯ r+ r0 r r1 )r rÔ rè Úz)
|
r r rt r r r? r{ r¹ r- r5 )r$ rÊ r% rª r& r& r' Útest_objview_record¶ s
|
z TestRecord.test_objview_recordc C s: t jddtdfgd}d|d _t|d jt d¡ d S )Nr0 rÔ r3 r+ r )r r r- rÔ r r )r$ r r& r& r' Útest_record_scalar_setitemÀ s
|
z%TestRecord.test_record_scalar_setitemc s0 t jddtfdtfgd tt fdd d S )N)r1 rÔ rè r+ c s
|
dg S )NZnofieldr& r& ©r¦ r& r' ré É rÈ z/TestRecord.test_missing_field.<locals>.<lambda>)r rt r5 r ÚKeyErrorrÙ r& r r' Útest_missing_fieldÆ s z TestRecord.test_missing_fieldc C s, t d¡t jdddgdg}t j |¡}d S )Nrq )r) z<u2)r* z<f4r+ )r Zaranger r r_ )r$ Zarraysr¦ r& r& r' Ú(test_fromarrays_nested_structured_arraysË s þz3TestRecord.test_fromarrays_nested_structured_arraysÚnfieldsr r0 r c C s| t dt jfdt jfdt jfgd |
¡}t d|¡ t j¡}|jjt jksPt|jt jks`t||_|jjt jksxtd S )Nr) r* rG r1 ) r r Úuint8rt r r r r rý )r$ r rÊ r6 r& r& r' Útest_assign_dtype_attributeÒ s *z&TestRecord.test_assign_dtype_attributec C s¢ t dt jfdt jfdt jfgd|
¡}t d|fg¡}t d|¡ t j¡}t|t js^tt|d t jsrt|d }t|t jstt|d t jstdS )z> Test that nested structured types are treated as records too r) r* rG NÚinnerr1 r ) r r r rt r r r´ rý r )r$ r rÊ Údt_outerr6 Údata0r& r& r' Ú test_nested_fields_are_recordsß s *z)TestRecord.test_nested_fields_are_recordsc C s t dt jfdt jfdt jfg¡}|ddg }|j|jks>tt d|fg¡}t d|¡ t j¡}t|d j| |d }t|d j| dS )z) test that trailing padding is preserved r) r* rG r r1 r N) r r r r" rý rt r r r )r$ rÊ Z dt_padded_endr r6 r r& r& r' Útest_nested_dtype_paddingí s " z$TestRecord.test_nested_dtype_paddingN)r r r rÚ rÝ rÞ rá rå rê rò ró rö r r r r
|
r r ÚmarkZ parametrizer r r r& r& r& r' rÖ S s&
|
|
|
|
rÖ c C s¦ ddddddg} t tj | ¡g k ddddddg}t tj |¡dgk ddddddddg}t tj |¡ddgk ddddddddg}t tj |¡ddgk d S )Nr0 r r1 r2 r3 r4 )r r r Zfind_duplicate)Úl1Úl2Úl3r& r& r' Útest_find_duplicateý s r )Úcollections.abcrµ r Úior Úosr Úpathlibr r r r Z numpy.testingr r r r r r
|
Z numpy.compatr r rÌ rÖ r r& r& r& r' Ú<module> s 5 +
|
