3
���h�D � @ s� d dl Zd dlmZmZmZ d dlZd dlmZ dd� Z dd� Z
dd � Z ej j
d
d �d
d� �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� ZG dd � d �ZdS ) � N)�assert_array_equal� assert_equal�
assert_raises)�chainc C s� dddgdddggdddgdddggg} xZdD ]R}t j| |d�}t j|d
d�}t|jt j� t|t jdgdggdgd ggg�� q.W ttt jt j| t d�� d S ) N� r z ?bBhHiIlLqQ)�dtype)�axis� �@ � � ���)
�np�array�packbitsr r �uint8r r � TypeError�float)�a�dt�arr�b� r �>/tmp/pip-build-5_djhm0z/numpy/numpy/lib/tests/test_packbits.py�
test_packbits s
(r c
C sf dddd d
d d d
g} xLdD ]D}x>| D ]6}t j||d�}t j|�}t|jt j� t|jd� q$W qW d S )Nr �
� z ?bBhHiIlLqQ)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 �emptyr r r r �shape)�shapesr r r r r r r �test_packbits_empty s
r c
C s� d d
gfd d d
dgfddddgfddddgfddddgfdd d d gfd d d!d"gfd#d$d%d&gfg} xpdD ]h}x`| D ]X\}}xLt |�D ]@\}}tj||d�}tj||d�}t|jtj� t|j|� � q�W � qpW � qfW d S )'Nr r r � � z ?bBhHiIlLqQ)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 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 )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
r' �bitorder�little�bigc - s
t jdddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd�g,�}|jd�}�x 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 k�r�t||� tt j|| d!�d dt� |� � fd#d$�t dd%�D �}t|dddd d&dd'd%dddd d(d)d g� � j
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 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� �qvW x�drD ]z}t j||d�� t j
jt j|�jt j|�j� j|ds�}d||dk< � |j|�9 � t j� dud,�}tt j|�d dv� |� � qxW ttt jt j|td�� d S )wNr r r" z ?bBhHiIlLqQ)r )r r( � � r � � � �� � � �q � � �~ �� �� � �? �8 � � �p �� �� � � r* )r( � c s( g | ] }t j� d | � d d�d �qS )N)r r 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 r �repeatr r r r r �
unpackbits�rangeZreshape�T�copy�random�randintZiinfo�min�maxr� Zastyper r r )r( r r r �rZrndr )r r �test_packbits_large4 s" ,(((((((((((((0
"
r� c C sF x@t dd�D ]2} x,dD ]$}tjd| ftd�}tj|dd� qW q W d S )Ni� i z ?bBhHiIlLqQ�� )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 jddddddddgddddddddgddddddddgg�� 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 r) )r r( r* r� )r( r 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 j| �}t|jt j� t|t jd�� d S )Nr )r )r )r )r r r r� r r r )r r r r r �test_unpackbits_empty� s
r� c C s� dgd fd
d d gd
fdddgdfdddgdfdddgdfddd gd fd d d gd!fd"d#d$gd%fg} xb| D ]Z\}}xNt |�D ]B\}}tj|tjd�}tj||d�}t|jtj� t|j|� � qxW � qfW d S )&Nr r! rI rG r" )r )r )r )r )r! rI r )rG r" r )rG rI r )rG rI r )r! r rI )rG r rI )rG r r" )rG r rI )r rG rI )r r! rI )r rG r" )r rG rI )r" r r )rI r r )rI r r )rI r r )r rI r )r r" r )r rI r )r rI r )r r rI )r r rI )r r r" )r r rI )r r r )r r r )r r r )r r r )r# r r r r� r r r )r Z in_shapesr& r% r$ r r r r r � test_unpackbits_empty_with_axis� s r� c C s� t jdt jd�} tt jt j| ��| � tt jt j| d d d� ��| d d d� � t j| d�} tt jt j| dd�dd�| � | jj� } tt jt j| dd�dd�| � d S ) Ni )r 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 Zejdddddddgdddddddgdddddddgdddddddgdddddddgdddddddgdddddddggejd�Zejdejd�Zej � edd�< ejdejd�Z
eddd� j � j � e
dd�< ejd ejd�Z ee dd�dd�f< e
jjd d �e
jjd eed
�ed d d ���dd� ��Ze
jjdi d dig�dd� �Ze
jjd d!�e
jjd eed�ed"d#d$���dd� ��Ze
jjdi d did d
id
dd�d d id dd�g�dd� �Zdd� ZdS )%� TestCountr r )r �9 N�1 � r. r( r) r* �count�: c C s\ |dk r|d }n|}t j| j|d�}t j|||d�}t|jt j� t|| jd |� � d S )Nr r )r( )r� r( ) r r r� r� r r r r �padded1)�selfr( r� �cutoff�packed�unpackedr r r �test_roundtrip. s
zTestCount.test_roundtrip�kwargsc C s@ t j| j�}t j|f|�}t|jt j� t|| jd d� � d S )Nr r
) r r r� r� r r r r r� )r� r� r� r� r r r �
test_count< s zTestCount.test_countrZ 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 r� r( )
r r r� r� r r r r �padded2r )r� r( r� r� �packed0� unpacked0�packed1� unpacked1r r r �test_roundtrip_axisE s
$
z TestCount.test_roundtrip_axis)r( r� c C sF t j| jdd�}t j|fddi|��}t|jt j� |jdd�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� |jdd�dk�r 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 r( r* r r
r
r
r
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|dd d� t j| j�}ttt j|d
d� d S ) Nr )r r� )r r� r r� )r� i����i����i���)r r r� r r� r� )r� r� r� r� r r r �test_bad_countr s zTestCount.test_bad_countr
)r� r� )r) r* r
i���r
)r) r* r
i����r
)�__name__�
__module__� __qualname__r r r r� �zerosr� ZravelZpadded1br� r� �pytest�mark� parametrizer r� r� r� r� r� r� r r r r r� s8
*
*r� )r) r* )Znumpyr Z
numpy.testingr r r r� � itertoolsr r r r' r� r� r� r� r� r� r� r� r� r� r r r r �<module> s
# |