U
|
P ±dF ã @ s d dl Zd dlmZmZmZ d dlZd dlmZ dd Z dd Z
|
dd Z ej d
|
d ¡d d Zdd Zdd Zdd Zdd Zdd Zdd ZG dd dZdS ) é N)Úassert_array_equalÚ assert_equalÚ assert_raises)Úchainc C s dddgdddggdddgdddggg} dD ]R}t j| |d}t j|dd}t|jt j t|t dgdggd gd
|
ggg¡ q,ttt jt j| t d d S ) Né r ú ?bBhHiIlLqQ©Údtypeéÿÿÿÿ©Úaxisé é@ éÀ é )
|
ÚnpÚarrayÚpackbitsr r Úuint8r r Ú TypeErrorÚfloat)ÚaÚdtÚarrÚb© r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_packbits.pyÚ test_packbits s ÿ&r c C s^ ddddddddg} d D ]@}| D ]6}t j||d
|
}t |¡}t|jt j t|jd q qd S ) N©r ©é
|
é r ©r r r! ©r r r! ©r! r r ©r r! r ©r r r! ©r r r r r )r Úemptyr r r r Úshape)Úshapesr r) r r r r r Útest_packbits_empty s þ
|
r+ c C sÄ ddgfddddgfddddgfddd d
|
gfd d d d gfd d dd gfddddgfddddgfg} dD ]Z}| D ]P\}}t |D ]>\}}tj||d}tj||d}t|jtj t|j| q|qlqdd S )Nr r )é r! r )r é r r" )r, r r! )r r r- r# )r r, r! )r r r- r$ ©r- r r r% ©r r- r r& ©r r r- r' r r r )Ú enumerater r( r r r r r) )r* r Úin_shapeZ
|
out_shapesÚaxÚ out_shaper r r r r Ú test_packbits_empty_with_axis s ø
|
r5 Úbitorder©ÚlittleÚbigc . s
|
t ddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddg,¡}| d¡}dD ]ö}t j||d t j d | d}t|jt j ddd dd
|
d ddd d d dddddddddd ddd ddddddddd d ddddddd dddddddddddd ddd ddddddd dddd d d ddd dddddddddddd dddddd ddddddddd dddddddd ddd d d dgq}| d krt|| tt j|| d!d d"
| fd#d$t dd%D }t|dddd d&dd'd%dddd d(d)d g
|
d*d+¡ t j dd,}t|jt j t|d-d.d/d/d0d1d2d3d3d4d d5d6d6d7d7d8d8d9d:d;d<d=d>d?gd@dAdd dd4dBdCdCd3d3dDddEdFdGdHdIdJdKdLdMdNdCdOgddPddAdQd'dRdSdTd0d0d0dUdVdVdWdddXdYdZd[d[d\d]gd^dVdVdGd_d`d`dd@d?dadadbdcdddedfdgdhdididBdjdkdlgdddd dmddndododd dHdpddodd dHdpdpdqd dd dgg t j dd,}t|jt j t|ddd dgd dddgd dddgddddgd d ddgddddgdd d dgd dddgddddgddd dgddddgddddgddddgddddgddddgddddgddd dgd d ddgdd d dgddd
|
dgddddgdd ddgddd dgddddgddddgddddgdd ddgddddgddddgddddgddddgd dddgddddgddd dgd dddgddddgg$ j ¡ t j dd,}t|jt j t|dd d dd ddd dddddddddd dddddddddddddd ddd dg$ddddd dd ddddddddddd d ddd ddddd dddddddddg$d dddddd ddd ddddddd dd d
|
ddd ddddddddddd ddg$ddddddddddddddddddddddddddddddddddddg$g t j dd,}t|jt j t|d-d@dd^dgd.dAdPdVdgd/dddVdgd/d dAdGd gd0ddQd_dmgd1d4d'd`dgd2dBdRd`dngd3dCdSddogd3dCdTd@dogd4d3d0d?dgd d3d0dad gd5dDd0dadHgd6ddUdbdpgd6dEdVdcdgd7dFdVdddogd7dGdWdedgd8dHddfd gd8dIddgdHgd9dJdXdhdpgd:dKdYdidpgd;dLdZdidqgd<dMd[dBd gd=dNd[djdgd>dCd\dkd gd?dOd]dldgg qrdrD ]z}t j||d t j jt |¡jt |¡j j|ds}d||dk< | |¡9 t j dtd,}tt |¡d d"
| qpttt jt j|td d S )uNr r r- r r ©r r6 éü é r éþ é é éð é éÿ éq éø é é~ éà éã é é? é8 é é ép éñ éÇ é é r9 ©r6 éüÿÿÿc s( g | ] }t j d |
d dd qS )Nr r
|
)r r )Ú.0Úi©r r r Ú
|
<listcomp>Y s z'test_packbits_large.<locals>.<listcomp>é é é éÆ éÄ é$ é r é¾ éº é² é é× éW éS éà éÎ éÌ é é é é( éi ék éK éJ éX éH éØ éÊ éZ éw ém éI r éÐ éô é½ é- é) éh éz é éx é é< é4 é¶ é éÒ éö éß é é é é é£ éÖ éD é é\ éN én é' éµ é éÜ éÞ éÚ éê éª é¨ éP é0 r é é° Z
|
bBhHiIlLqQ)ÚlowÚhighÚsizer r
|
)r r Úrepeatr r r r r Ú
|
unpackbitsÚrangeZreshapeÚTÚcopyÚrandomÚrandintZiinfoÚminÚmaxr¥ Zastyper r r )r6 r r r ÚrZrndr rW r Útest_packbits_large4 sD , ò
|
|
ù
|
|
ÿ þ þ þ þ þô
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Ý%
|
ý ý ý þô è
|
|
þ r° c C s> t ddD ].} dD ]$}tjd| ftd}tj|dd qq
|
d S )Ni¶ i r éÈ r r r )r¨ r ZonesÚboolr )Úsr Úxr r r Útest_packbits_very_largeÖ s rµ c C s t jdgdgdggt jd} t j| dd}t|jt j t|t ddddddddgddddddddgddddddddgg¡ d S )Nr, r> é r r r r )r r r r§ r r r ©r r r r r Útest_unpackbitsß s þr¸ c C sÒ t jdgdgdggt jd} t j| dd}t|jt j t j| ddd}t j| dd d}t|| t| t j|ddd t|d d
d d d
|
f | t| t j|dd d tt t j| d d tt
|
t j| d d d S )Nr, r> r¶ r r r r8 r: r9 r
|
r¯ rS r ) r r r r§ r r r r r Ú
|
ValueErrorr )r r Zb_littleZb_bigr r r Útest_pack_unpack_orderè s
|
rº c C s< t jdt jd} t | ¡}t|jt j t|t d¡ d S )Nr r )r r( r r§ r r r r· r r r Útest_unpackbits_empty÷ s
|
r» c
|
C s¼ dgdfdddgdfdddgdfdd d
|
gdfd d d gd fd dd gd fdddgdfdddgdfg} | D ]R\}}t |D ]@\}}tj|tjd}tj||d}t|jtj t|j| qtqdd S )Nr )r, r[ r )rY r- r )rY r[ r )r, r r[ )rY r r[ )rY r r- )r rY r[ )r r, r[ )r rY r- r. )r[ r r )r r[ r r/ )r r r[ r0 r' r r )r1 r r( r r§ r r r) )r* Z in_shapesr4 r3 r2 r r r r r Ú test_unpackbits_empty_with_axisþ s ø
|
r¼ c C s¦ t jdt jd} tt t | ¡¡| tt t | d d d
¡¡| d d d
t | d¡} tt jt j| dddd| | j ¡ } tt jt j| dddd| d S )Ni r r, )r- r r r r ) r Zaranger r r r§ Ztiler© rª )Údr r r Útest_unpackbits_large s *
|
r¾ c @ sÊ e Zd ZejdddddddgdddddddgdddddddgdddddddgdddddddgdddddddgdddddddggejdZejdejdZe ¡ edd
< ejdejdZ
|
eddd
¡ ¡ e
|
dd
< ejdejdZ ee dd
dd
f< e j d
|
d ¡e j d eed eddd¡dd Ze j di d dig¡dd Ze j d
|
d ¡e j d eededdd¡dd Ze j di d did
|
didddd
|
didddg¡dd Zd d ZdS ) Ú TestCountr r r é9 Né1 r
|
)é r r> r6 r7 Úcounté: éÇÿÿÿc C s\ |dk r|d }n|}t j| j|d}t j|||d}t|jt j t|| jd |
d S )Nr r rS )rà r6 © r r r´ r§ r r r r Úpadded1)Úselfr6 rà ÚcutoffÚpackedÚunpackedr r r Útest_roundtrip. s
|
zTestCount.test_roundtripÚkwargsc C s@ t | j¡}t j|f|}t|jt j t|| jd d
d S )Nr
|
rÆ )rÈ rÍ rÊ rË r r r Ú
|
test_count< s zTestCount.test_countrl é÷ÿÿÿc C sÆ |dk r|d }n|}t j| jd|d}t j|d||d}t|jt j t|| jd |
d | jj d
f t j| jd|d}t j|d||d}t|jt j t|| jd | jj d
d |
f d S )Nr r r: )r rà r6 )
|
r r r´ r§ r r r r Úpadded2r) )rÈ r6 rà rÉ Úpacked0Ú unpacked0Úpacked1Ú unpacked1r r r Útest_roundtrip_axisE s
|
|
ÿ$
|
ÿz TestCount.test_roundtrip_axisr8 )r6 rà r9 c C sF t j| jdd}t j|fddi|}t|jt j | dd¡dkrjt|| j d d
d | jj
|
d
f n6t|d d d
d d
f | j d d
d | jj
|
d
f t j| jdd}t j|fddi|}t|jt j | dd¡dkr t|| j d | jj
|
d
d d
f n6t|d d
d d d
f | j d | jj
|
d
d d
f d S )Nr r r r6 r9 r
|
r ) r r r´ r§ r r r Úgetr rÐ r) )rÈ rÍ rÑ rÒ rÓ rÔ r r r Útest_axis_countZ s &6&zTestCount.test_axis_countc C sj t j| jdd}ttt j|ddd t j| jdd}ttt j|ddd t | j¡}ttt j|dd d S )Nr r rÏ )r rà r rÅ )rà )r r r´ r r¹ r§ )rÈ rÑ rÓ rÊ r r r Útest_bad_countr s zTestCount.test_bad_count)Ú__name__Ú
|
__module__Ú __qualname__r r r r´ ÚzerosrÇ ZravelZpadded1brª rÐ ÚpytestÚmarkÚ parametrizer r¨ rÌ rÎ rÕ r× rØ r r r r r¿ sL ùø ÿ
|
û
|
r¿ )Únumpyr Z numpy.testingr r r rÝ Ú itertoolsr r r+ r5 rÞ rß r° rµ r¸ rº r» r¼ r¾ r¿ r r r r Ú<module> s
|
"
|
