%PDF- %PDF-
Mini Shell

Mini Shell

Direktori : /lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/
Upload File :
Create Path :
Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_packbits.cpython-39.opt-1.pyc

a

z[yc�D�@s�ddlZddlmZmZmZddlZddlmZdd�Z	dd�Z
dd	�Zej�
d
d�dd
��Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�ZGdd�d�ZdS)�N)�assert_array_equal�assert_equal�
assert_raises)�chaincCs�gd�gd�ggd�gd�gg}dD]R}tj||d�}tj|dd�}t|jtj�t|t�d	gd
ggdgdggg��q$tttjtj|t	d��dS)
N)�rr)rrr)rrr)rrr�?bBhHiIlLqQ��dtype�����axis��@�� )
�np�array�packbitsrr	�uint8rr�	TypeError�float)�a�dt�arr�b�r�C/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_packbits.py�
test_packbitss�&rcCsRgd�}dD]@}|D]6}tj||d�}t�|�}t|jtj�t|jd�qqdS)N)�r��
�r�r rr!�rr r!�r!rr�rr!r�rrr!�rrrrrr)r�emptyrrr	r�shape)�shapesrr)rrrrr�test_packbits_emptys
r+c
Cs�ddgfdgd�fdgd�fdgd�fdgd	�fd
gd�fdgd
�fdgd�fg}dD]Z}|D]P\}}t|�D]>\}}tj||d�}tj||d�}t|jtj�t|j|�qnq^qVdS)Nrr))�r!r)r �rrr"))r,rr!r")r rr-r#)r#)rr,r!)rr r-r$)�r-rrr$r$r%)r%�rr-rr%r&)r&r&�rrr-r'�r'r'r'rrr)�	enumeraterr(rrr	rr))r*r�in_shapeZ
out_shapes�ax�	out_shaperrrrr�test_packbits_empty_with_axis s 






�
r6�bitorder��little�bigc(s�t�gd��}|�d�}dD�]�}tj||d��tj�d|d�}t|jtj�gd�}|dkrht||�ttj||d�dd	�|��fd
d�t	dd
�D�}t|gd����
dd��tj�dd�}t|jtj�t|gd�gd�gd�gd�gd�g�tj�dd�}t|jtj�t|gd�gd�gd�gd�gd�gd�gd�gd�gd �gd!�gd"�gd#�gd$�gd%�gd&�gd'�gd(�gd)�gd*�gd+�gd �gd,�gd-�gd.�gd/�gd0�gd1�gd2�gd3�gd4�gd5�gd6�gd7�gd8�gd9�gd:�g$��j���tj�dd�}t|jtj�t|gd;�gd<�gd=�gd>�g�tj�dd�}t|jtj�t|gd?�gd@�gdA�gdB�gdC�gdD�gdE�gdF�gdG�gdH�gdI�gdJ�gdK�gdL�gdM�gdN�gdO�gdP�gdQ�gdR�gdS�gdT�gdU�gdV�gdW�g�qdXD]z}tj||d��tj
jt�|�jt�|�j�j|dY�}d||dk<�|�|�9�tj�dZd�}tt�|�dd	�|��qtttjtj|td��dS)[N(,rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr-rr�rr7)q��rr-��r<rr?���r�r�r<�q�r-rCrrB�rrB�~r��r=rC���r?r@rJ�?rBrG�8rIrArrI�rKrH�rLr<�prLrC��rEr-rArLrHrOrKrCrC��rHrNrr@�rr-rCrQrrrCrHrrCr<rGrKrrrr<rNrKrrFrQr<rDrCr-rMrLr<rNr?rrDrCrCrJrLrI�rErIrSrQr@rMr:�r7���cs(g|] }tj�d|�dd�d�qS)Nrr
)rr)�.0�i�rrr�
<listcomp>Y�z'test_packbits_large.<locals>.<listcomp>r�)rMrMrMr@�rB�r[rrrrQ����r�$�rr)���rd����W�Srh��rQ����rk�rl�rm��(�i�k�K�J�X)�H��rErPrIri���Zrxrhrh�wr=�m�Ir������-�)�h�zrx�)rD�xrErv�r]�<�4�rerere���r��rCrC�����r���)��r�r�r�D�r�rrurt�\r��N�n�'��������r�rw����)rrMrMr�PrO�0r
r
rHrAr|�rMr
rHrAr|r�r��rArHrrM)r<r=rr)r?r<rFrM)rArrBr)rCrMrrM)rr@rCrM)rJrKrr)rCrAr?r)r?rHrNr)rGrrHr)rCrCrQr)rLrBrGr)rDrErIrM)rIrJrKr)rrBrOr)rFrEr-rM)rBrGrLr)rLrCrPrM)rAr?rHr)rIrSrrM)rCrCr>r)r-rPrEr)rrCrPrM)rMrrCrM)rHrrCrM)rEr<rGr)rr?r-rM)rHrDrEr)rr<r=rM)rJrKrHr)rHrNrKr)r?r-rMr)rDrCrCrM)rBrDrQr)r?rIrJr)rNrLr<r)$r<r?rArCrrJrCr?rGrCrLrDrIrrFrBrLrArIrCrGr-rrMrHrErrHrrJrHr?rDrBr?rN)$r=r<rrMr@rKrArHrrCrBrErJrBrErGrCr?rSrCrrPrCrrr<r?rDr<rKrNr-rCrDrIrL)$rrFrBrrCrr?rNrHrQrGrIrKrOr-rLrPrHrr>rHrErPrCrCrGr-rEr=rHrKrMrCrQrJr<)$rrMrrMrMrrrrrrrMrrrMrrMrrMrrrrMrMrMrrMrrMrrrrMrrr)rbrurDr�r)rcrvr�r�rM)rdrErEr�rM)rdrPrvrr)rerIr�r�r�)rfrir]r�rO)rgrwr�r�r�)rhrxr�rr
)rhrxr�rur
)rirhrertrH)rQrhrer�rA)rjryrer�r|)rkr=r�r�r�)rkrzr�r�rM)rlr{r�r�r
)rlrr�r�rH)rmr|rCr�rA)rmr}rCr�r|)rnr~r�r�r�)rorr�r�r�)rpr�r�r�r�)rqr�r�rwrA)rrr�r�r�rH)rsrxr�r�r)rtr�r�r�rMZ
bBhHiIlLqQ)ZlowZhigh�sizer	r
)rr�repeatrrr	rr�
unpackbits�rangeZreshape�T�copyZrandomZrandintZiinfo�min�maxr�Zastyperrr)r7rr	r�rZrndrrXr�test_packbits_large4s�



�
�%

�
��r�cCs>tdd�D].}dD]$}tjd|ftd�}tj|dd�qq
dS)Ni�ir��rrr)r�rZones�boolr)�sr�xrrr�test_packbits_very_large�sr�cCs^tjdgdgdggtjd�}tj|dd�}t|jtj�t|t�gd�gd�gd	�g��dS)
Nr,r?�rrr)rrrrrrrr)rrrrrrrr)rrrrrrrr)rrrr�rr	r�rrrrr�test_unpackbits�s�r�cCs�tjdgdgdggtjd�}tj|dd�}t|jtj�tj|ddd�}tj|dd	d�}t||�t|tj|ddd��t|dd�ddd
�f|�t|tj|dd	d��tt	tj|dd�tt
tj|d
d�dS)Nr,r?r�rrrr9r;r:r
r�rTr )rrrr�rr	rrr�
ValueErrorr)rrZb_littleZb_bigrrr�test_pack_unpack_order�s
r�cCs<tjdtjd�}t�|�}t|jtj�t|t�d��dS)Nrr)rr(rr�rr	rr�rrr�test_unpackbits_empty�s
r�c	Cs�dgdfgd�dfgd�dfgd�dfgd�d	fgd
�dfgd�d
fgd�dfg}|D]R\}}t|�D]@\}}tj|tjd�}tj||d�}t|jtj�t|j|�qfqVdS)Nr))r,r]r)r[r-r�r[r]rr�))r,rr]�r[rr])r[rr-r�)�rr[r])rr,r])rr[r-r�)r.�r]rrr�r�)�rr]rr/r�r�)�rrr]r�r0r�r1r'rr)r2rr(rr�rr	r))r*Z	in_shapesr5r4r3rrrrr�test_unpackbits_empty_with_axis�s






�
r�c	Cs�tjdtjd�}tt�t�|��|�tt�t�|ddd���|ddd��t�|d�}ttjtj|dd�dd�|�|j��}ttjtj|dd�dd�|�dS)Nirr,)r-rrrr)	rZarangerrrr�Ztiler�r�)�drrr�test_unpackbits_larges*
r�c@s�eZdZejgd�gd�gd�gd�gd�gd�gd�gejd�Zejd	ejd�Ze�	�ed
d�<ejd	ejd�Z
ed
d
d����	�e
d
d�<ejd
ejd�Zeed
d�d
d�f<e
j�dd�e
j�deed�eddd���dd���Ze
j�didd
ig�dd��Ze
j�dd�e
j�deed�eddd���dd���Ze
j�didd
iddidd
d�ddidd
d�g�d d!��Zd"d#�Zd
S)$�	TestCount)rrrrrrr)rrrrrrr)rrrrrrr)rrrrrrr)rrrrrrr)rrrrrrr)rrrrrrrr�9N�1r
)�	r�r?r7r8�count�:����cCs\|dkr|d}n|}tj|j|d�}tj|||d�}t|jtj�t||jd|��dS)NrrrT)r�r7�	rrr�r�rr	rr�padded1)�selfr7r��cutoff�packed�unpackedrrr�test_roundtrip.s
zTestCount.test_roundtrip�kwargscCsDt�|j�}tj|fi|��}t|jtj�t||jdd��dS)Nr
r�)r�r�r�r�rrr�
test_count<szTestCount.test_countrn���cCs�|dkr|d}n|}tj|jd|d�}tj|d||d�}t|jtj�t||jd|�d|jj	d�f�tj|jd|d�}tj|d||d�}t|jtj�t||jd|jj	d�d|�f�dS)Nrrr;)rr�r7)
rrr�r�rr	rr�padded2r))r�r7r�r��packed0�	unpacked0�packed1�	unpacked1rrr�test_roundtrip_axisEs

�$
�zTestCount.test_roundtrip_axisr9)r7r�r:cCsFtj|jdd�}tj|fddi|��}t|jtj�|�dd�dkrjt||j	dd�d|jj
d�f�n6t|ddd�dd�f|j	dd�d|jj
d�f�tj|jdd�}tj|fddi|��}t|jtj�|�dd�dk�rt||j	d|jj
d�dd�f�n6t|dd�ddd�f|j	d|jj
d�dd�f�dS)Nrrrr7r:r
r)rrr�r�rr	r�getrr�r))r�r�r�r�r�r�rrr�test_axis_countZs&6&zTestCount.test_axis_countcCsjtj|jdd�}tttj|ddd�tj|jdd�}tttj|ddd�t�|j�}tttj|dd�dS)Nrrr�)rr�rr�)r�)rrr�rr�r�)r�r�r�r�rrr�test_bad_countrszTestCount.test_bad_count)�__name__�
__module__�__qualname__rrrr�Zzerosr�ZravelZpadded1br�r��pytest�mark�parametrizerr�r�r�r�r�r�rrrrr�sH��	�
�
r�)ZnumpyrZ
numpy.testingrrrr��	itertoolsrrr+r6r�r�r�r�r�r�r�r�r�r�rrrr�<module>s 


"		

Zerion Mini Shell 1.0