U
|
P ±dO ã @ sÆ d Z ddlZddlmZ ddlmZ ddlmZmZ ddlm Z ddl
|
m Z m Z ddlmZmZm Z mZm Z mZ ddlmZmZmZ dd lmZ G d
|
d d ZG d d d ZG dd dZ dd Z dS )z^Tests suite for mrecords.
|
|
:author: Pierre Gerard-Marchant
|
:contact: pierregm_at_uga_dot_edu
|
|
é N)Úrecarray)ÚmaskedÚnomask)Útemppath)Ú fromrecordsÚ
|
fromarrays)Ú MaskedRecordsÚ mrecarrayr Ú fromtextfiler Úaddfield)Úassert_Ú assert_equalÚassert_equal_records)Úpicklec @ sê e Zd ZdddddgZdddd d
|
gZd d d ddgZdefdefdgZdddddgZ e
|
j e e eeee edZdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Z d1d2 Z d3S )4Ú TestMRecordsé é é é é çñ?ç@çffffff
|
@ç@ç @ó oneó twoó threeó fouró fiveÚaÚb©Úcú|S8r ©ÚmaskÚdtypec C s | j }| t¡}t|j|j t|j|j tt|j t
|
t|j |j t
|
¡ dD ]}t|| || qTt| t¡| d S )N©r r! r# ) ÚbaseÚviewr r Ú
|
recordmaskr Ú_maskr Ú
|
isinstanceÚ_datar )Úselfr) ÚmbaseÚfield© r2 úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/ma/tests/test_mrecords.pyÚ test_byview$ s
|
zTestMRecords.test_byviewc C s¶ | j ¡ }| t¡}dD ]*}tt|||| t|| || q|d }tt|t t|j|j t| ¡ d t|j
|
t t|j ¡ d t|d |d d |d }tt|t t|j|j t| ¡ d t|j
|
d t|j ¡ d t|d |d d t|d tk |d d
|
}tt|t t|j|j t|j
|
dd g t|jtjdd g|j jd t||d d
|
t¡ dD ]"}tt|||d d
|
| qd S ) Nr( r )r r r )FFFr éÿÿÿÿ)NNNT)TTTr r ©r' )r) Úcopyr* r r Úgetattrr r- r' Útolistr+ r r, Úitemr r r& ÚnpÚarray)r/ r) r0 r1 Z mbase_firstZ
|
mbase_lastZmbase_slr2 r2 r3 Útest_get0 sD
|
|
ÿþÿzTestMRecords.test_getc C sH | j ¡ }| t¡}| ¡ }d|_d|jjd d
< t|d jdddddg t|d jdddddg d|_t|d jdgd tt
|
|d ¡dgd t|j dgd t|j ¡ t jdddddgtd t|_t|jjdgd t|jj dgd tt
|
|d
|
¡dgd tt |d
|
¡d gd t|j ¡ t jd dd d dgtd | t¡ ¡ }d|jd d
< t|jddd ddg t|jjdddddg t|jd d
< t|j|d t|jjdddddg ddtfg}t jdddg|d }| t¡}t jdddgdddgd}||d< t|j| t|jjdddg d S )N)i?B g@µx ¯DúN/Ar r r r F©r r r )r r r r6 r# ó N/A)r r r r r r! )Úalphaz|S1Únum)r r )r! r )r# r é
|
é é ©r& )r) r7 r* r Ú
|
fill_valuer r. r r, ÚmaÚ getmaskarrayr+ r9 r; r< Úboolr r# r& Zgetdatar! Úintr rB )r/ r) r0 ÚndtypeÚdataZrdataÚvalr2 r2 r3 Útest_set_fieldsX sd
|
|
|
üûÿ
|
üûÿ
|
z TestMRecords.test_set_fieldsc C s¶ | j ¡ }| t¡}t|d d< t|jdddddg t|jjdddddg tt
|
d¡t j d¡gdt fd tfgd
|
}t|d d< t|jdddddg t|jjdddddg d S ) Nr éþÿÿÿr r r r r r r! r6 )r) r7 r* r r r r r, r r; ÚarangeÚrandomÚrandrK Úfloat©r/ r) r0 r2 r2 r3 Útest_set_fields_mask s
|
|
ÿ z!TestMRecords.test_set_fields_maskc C sÄ | j ¡ }| t¡}t|_tt |d ¡dgd t|d j |d j t|d j |d j t|j
|
¡ t j dgd t d t|_tt |d ¡dgd t|j
|
¡ t j d gd t d d S )
|
Nr! r r r r# ©r r r r6 r r? )r) r7 r* r r r& r rH rI r, r9 r; r< rJ r rU r2 r2 r3 Ú test_set_mask s
|
|
|
ÿ
|
ÿzTestMRecords.test_set_maskc C sÈ | j ¡ }| t¡}dddddg|_t|jjdddddg t|jjdddddg t|jjdddddg dddddg|_t|jjdddddg t|jjdddddg t|jjdddddg d S )Nr r ) r) r7 r* r r& r r r! r# rU r2 r2 r3 Útest_set_mask_fromarray® s
|
|
z$TestMRecords.test_set_mask_fromarrayc C sà | j ¡ t¡}tjdddddgdtfdtfdtfgd}||_t|j jddd d dg t|j
|
jd d dddg t|j jddd d dg d
|
|_||_ t|j jddd d dg t|j
|
jd d dddg t|j jddd d dg d S ) N)r r r )r r r r? r r! r# r6 r r F) r) r7 r* r r; r< rJ r& r r r! r# Z fieldmask)r/ r0 Znmaskr2 r2 r3 Útest_set_mask_fromfields¼ s þz%TestMRecords.test_set_mask_fromfieldsc C s² | j ¡ }| t¡ ¡ }t|d< t|j ¡ tj dddddgt
|
d t|j dddddg | t¡ ¡ }d|d d
< t|j j d d d
|
d d g t|j jdddddg t|jj d d d ddg t|jjdddddg t|jj dddddg t|jjdddddg | t¡ ¡ }t|d d
< t|j j ddd
|
d d g t|j jdddddg t|jj ddd ddg t|jjdddddg t|jj dddddg t|jjdddddg d S )NrP r? rW r6 r r ©r r r r r r r g @r r r ó 5r r r r r r r )r) r7 r* r r r r, r9 r; r< rJ r+ r r. r! r# rU r2 r2 r3 Útest_set_elementsÍ s<
|
ÿþ ÿ ÿz TestMRecords.test_set_elementsc C s | j ¡ }| t¡}| ¡ zd|dd
< t|jjdddddg t|jjddd dd
|
g t|j jd d d ddg t|jj
|
dddddg t|jj
|
|jj
|
t|jj
|
|j j
|
W n( t k
|
rÄ Y n t k
|
rØ Y n
|
X t dzd|dd
< W n t tfk
|
r Y n
|
X tdd S )Nr[ rP r r r r r r r r r r r r\ r r z)Flexible hard masks should be supported !z.Should have expected a readable buffer object!)r) r7 r* r Ú harden_maskr r r. r! r# r, ÚNotImplementedErrorÚAssertionErrorÚ ExceptionÚ TypeErrorrU r2 r2 r3 Útest_setslices_hardmaskí s.
|
|
ÿz$TestMRecords.test_setslices_hardmaskc C s | j ¡ }| t¡}| ¡ t|j t|_t |j
|
|j
|
| ¡ t|j t|_t |j
|
t |j|j¡ tt |d j
|
¡tk t|d j
|
|d j
|
d S )Nr! r )r) r7 r* r r^ r Z _hardmaskr r& r r, Z soften_maskrH Zmake_mask_noneÚshaper' Z make_maskr rU r2 r2 r3 Ú test_hardmask
|
s
|
|
|
ÿzTestMRecords.test_hardmaskc C s~ | j ¡ }| t¡}tdtjd D ]T}tj||d}t |¡}t |j
|
|j
|
t |j |j t |j |j t |j |j q$d S )Nr r )Úprotocol)r) r7 r* r Úranger ÚHIGHEST_PROTOCOLÚdumpsÚloadsr r' r r. r, )r/ r) ÚmrecÚprotoÚ_Zmrec_r2 r2 r3 Ú test_pickling s
|
|
|
zTestMRecords.test_picklingc C sÎ t jdddgdddgtd}t jdddgdddgtd}t jd d
|
d gdddgd d}d tfdtfdg}t|||g|dd}| ¡ }t|d tjdtd t|d tjdtd t|d tjdd d d S )Nr r r r r% r r r ÚoneÚtwoÚthreer$ r r! r" ©é ç ðiø@r> ©r' rG )r r rs r6 )r r rt r# )ro rp r> )rH r< rK rT r Úfilledr r; )r/ Ú_aÚ_bÚ_cÚddtyperk Z
|
mrecfilledr2 r2 r3 Ú test_filled' s ÿÿÿzTestMRecords.test_filledc C s t jdddgdddgtd}t jdddgdddgtd}t jd d
|
d gdddgd d}d tfdtfdg}t|||g|dd}t| ¡ dddg d S )Nr r r r r% r r r ro rp rq r$ r r! r" rr ru )r r N)r r r )NNr )rH r< rK rT r r r9 )r/ rw rx ry rz rk r2 r2 r3 Ú test_tolist6 s ÿÿÿzTestMRecords.test_tolistc C s0 t dtdd}d|d d< t|d d d d S )Nr r) )ÚformatsÚnamesrC r )r rT r )r/ Úxr2 r2 r3 Útest_withnamesC s zTestMRecords.test_withnamesc C sº t ddtfddtfgd}t|d< t| d¡ ¡ d t ddgd}t|d< t| d¡ ¡ tjd |j d ¡ t d
|
d d}t|d< d |d< | d¡ t
|
| d¡tjd d g|j d d S )Nr Úi)Úsr$ Úfr6 r )r ó 1g ð?)Zf0z<f8)r r )r r zi4, (2,3)float, floatrW r? ) r rK rT r r rv r: r; r< r' r )r/ ZeasyZsoloZmultr2 r2 r3 Útest_exotic_formatsI s" ÿ
|
|
|
ÿÿz TestMRecords.test_exotic_formatsN) Ú__name__Ú
|
__module__Ú __qualname__ZilistÚflistÚslistrK rT rz r& rH r< ÚlistÚzipr) r4 r= rO rV rX rY rZ r] rc re rn r{ r| r r
r2 r2 r2 r3 r s* (7 r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestViewc C sn t d¡t j d¡ }}dtfdtfg}t jtt|||d}t||g|dd}d|j d< ||||f| _
|
d S ) NrC r r! r6 )g "Àg ÀXÀru )FTr ) r; rQ rR rS rT r< r r r r& rM )r/ r r! rL Úarrrk r2 r2 r3 Ú setup_method_ s
|
zTestView.setup_methodc C s@ | j \}}}}| ¡ }tt|t t|| t|j|j d S )N)rM r* r r- r r r, )r/ rk r r! r Útestr2 r2 r3 Útest_view_by_itselfh s
|
|
z TestView.test_view_by_itselfc C sd | j \}}}}tdf}| |¡}tt|tj t|tj t
|
t ||td t|d tj k d S )Nr r6 )r r ) rM rT r* r r- rH Z MaskedArrayr r; r< r r r )r/ rk r r! r Úntyper r2 r2 r3 Útest_view_simple_dtypeo s
|
z TestView.test_view_simple_dtypec C s~ | j \}}}}dtfdtfg}| |¡}tt|t t|| |¡ t|d d tk t|j t
|
|¡ t|j d k d S )NÚAÚBr ) rM rT r* r r- r r r r r' r; Z _fill_value)r/ rk r r! r Zalttyper r2 r2 r3 Útest_view_flexible_typew s
|
z TestView.test_view_flexible_typeN)r r r r r r r r2 r2 r2 r3 r ] s r c @ sÐ e Zd ZejdddgdddgedZejdddgdddgedZejd d
|
d gdddgd dZ d efdefdgZ
|
e eee ge
|
ddZ e ejeje jfe
|
dZe ee
|
fZdd Zdd Zdd Zdd Zdd Zd S ) ÚTestMRecordsImportr r r r r% r r r r r r r$ r r! r" )s 99999s 99999.r@ ru r6 c
|
C sÆ t jdddgdddgtd}t jdddgdddgtd}t jd d
|
d gdddgd d}| j\}}}td |||fD ] \}}tt||j|j qpt jddd gdddgt d} t
|
t | |j d|d d S )Nr r r r r% r r r ro rp rq r$ r( r6 ) rH r< rK rT rM r r r8 r, Úobjectr r r' )
|
r/ rw rx ry rk Únrecrm r ÚlÚ_xr2 r2 r3 Útest_fromarrays s z"TestMRecordsImport.test_fromarraysc C s | j \}}}dddg}t|dd}t|dd}t|| t|}t|j|j |jjD ] }tt||t|j| qVt| ¡ dd}t|jdt
|
fdt fd g t d
|
d D ] \} }
|
tt|| t|j|
|
q¨t|}t|j|j t|j| ¡ t|j|j d S ) N)r Úabcg À @r )r Zxyg ÀÌÌ@r )r ú g Ù?r zc1, c2, c3, c4)r~ zc1,c2,c3Úc1Úc2)Úc3z|S5)r r r¡ r( )rM Úrecfromrecordsr r r r' r~ r8 r. r9 rK rT r rv r, ) r/ rk r rz ZpalistÚpaÚmpaÚ_mrecr1 r Únr2 r2 r3 Útest_fromrecords s( þ
|
z#TestMRecordsImport.test_fromrecordsc C sö | j \}}}t| ¡ |dddgd}t|j|j t|j ¡ dddg t| ¡ |dd}t|j|j t|j ¡ dddg t| ¡ ||jd}t|j|j t|j ¡ |j ¡ t| ¡ ||j ¡ d}t|j|j t|j ¡ |j ¡ d S )Nr r )r' r& r? rW T)rM r r9 r r. r r, )r/ rk r rz r¥ r2 r2 r3 Útest_fromrecords_wmaskµ s
|
ÿz)TestMRecordsImport.test_fromrecords_wmaskc
|
C s d}t 4}t|d}| |¡ W 5 Q R X t|ddd}W 5 Q R X tt|t t|jddddg t|j j
|
ddddg t|j dddd g d S )
|
Nz¿#
|
'One (S)','Two (I)','Three (F)','Four (M)','Five (-)','Six (C)'
|
'strings',1,1.0,'mixed column',,1
|
'with embedded "double quotes"',2,2.0,1.0,,1
|
'strings',3,3.0E5,3,,1
|
'strings',4,-1e-10,,,1
|
Úwú,ZABCDEFG)Ú delimiterÚvarnamesr r g OAg»½×Ùß|Û½) r ÚopenÚwriter
|
r r- r r ÚFÚEr, ÚC)r/ ZfcontentÚpathr Zmrectxtr2 r2 r3 Útest_fromtextfileÊ s ÿ z$TestMRecordsImport.test_fromtextfilec C sT | j \}}}dddgdddg }}t|tj||d}t|j| t|jj| d S )Néd éÈ i, r r rF )rM r rH r< r Zf3r, )r/ rk r rz ÚdÚmr2 r2 r3 Ú test_addfieldÝ s
|
z TestMRecordsImport.test_addfieldN)r r r rH r< rK rw rT rx ry rz r rk Ú recfromarraysr. r rM r r§ r¨ r³ r¸ r2 r2 r2 r3 r s" ÿ ÿ
|
r c C s0 t jddgddgdtfdtfgd} | d d S ) N)r Ú2)r Ú4)r r )r r r r! r% r )rH Z masked_arrayrK r )Úyr2 r2 r3 Ú#test_record_array_with_object_fieldæ s ýr½ ) Ú__doc__Únumpyr; Znumpy.marH r r r Z numpy.testingr Znumpy.core.recordsr r¢ r r¹ Znumpy.ma.mrecordsr r r
|
r Znumpy.ma.testutilsr r r Z numpy.compatr r r r r½ r2 r2 r2 r3 Ú<module> s D&c
|
