Edit File: test_packbits.cpython-37.pyc
B ��Fd�2 � @ s� d dl mZmZmZ d dlZd dlmZmZm Z m Z dd� Z dd� Z dd � Z d d � Zd d � Zdd� Zdd� Zdd� Zdd� Zedkr�e � dS )� )�division�absolute_import�print_functionN)�assert_array_equal� assert_equal� assert_raises�run_module_suitec C s� dddgdddggdddgdddggg} xZdD ]R}t j| |d�}t j|dd�}t|jt j� t|t �dgdggd gd 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 ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_packbits.py� test_packbits s (r c C sf ddddddddg} xLd D ]D}x>| D ]6}t j||d �}t �|�}t|jt j� t|jd� q$W qW d S ) N)r )� � r )r r r )r r r )r r r )r r r )r r r )r r r z ?bBhHiIlLqQ)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} xjdD ]b}x\| D ]T\}}xJt |�D ]>\}}tj||d�}tj||d�}t|jtj� t|j|� q�W qpW qfW d 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 )r r r% )r r r )r% r r )r r r )r r% r )r r r )r r r% )r r r z ?bBhHiIlLqQ)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* c . s t �dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd�g,�} | �d�} �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}t||� tt �|�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,d-d.d/d0d0d1d d2d3d3d4d4d5d5d6d7d8d9d:d;d<gd=d>dd dd1d?d@d@d0d0dAddBdCdDdEdFdGdHdIdJdKd@dLgddMdd>dNd%dOdPdQd-d-d-dRdSdSdTdddUdVdWdXdXdYdZgd[dSdSdDd\d]d]dd=d<d^d^d_d`dadbdcdddedfdfd?dgdhdigdddd djddkdldldd dEdmddldd dEdmdmdnd 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+d>dMdSdgd,dddSdgd,d d>dDd gd-ddNd\djgd.d1d%d]dgd/d?dOd]dkgd0d@dPddlgd0d@dQd=dlgd1d0d-d<dgd d0d-d^d gd2dAd-d^dEgd3ddRd_dmgd3dBdSd`dgd4dCdSdadlgd4dDdTdbdgd5dEddcd gd5dFddddEgd6dGdUdedmgd7dHdVdfdmgd8dIdWdfdngd9dJdXd?d gd:dKdXdgdgd;d@dYdhd gd<dLdZdidgg� �qvW x�doD ]z}t j| |d�� t j jt �|�jt �|�j� j|dp�}d||dk< � |�|�9 � t j� dqd�}tt �|�d d � | � � qhW ttt jt j| td�� d S )rNr r r% z ?bBhHiIlLqQ)r )r � � r � � � �� � � �q � � �~ �� �� � �? �8 � � �p �� �� � � ���c s( g | ] }t j� d | � d d�d �qS )N)r r )r r )�.0�i)r r r � <listcomp>[ s z'test_packbits_large.<locals>.<listcomp>� � � �� �� �$ � � � � � �� �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�range�reshape�T�copy�random�randint�iinfo�min�maxr� �astyper r r )r r r �rZrndr )r r �test_packbits_large8 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 �ones�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 jdt jd�} t �| �}t|jt j� t|t �d�� d S )N)r )r )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} x^| D ]V\}}xLt |�D ]@\}}tj|tjd�}tj||d�}t|jtj� t|j|� qxW qfW d S )N)r )r$ rI r )rG r% r )rG rI r )r$ r rI )rG r rI )rG r r% )r rG rI )r r$ rI )r rG r% )r% r r )rI r r )r rI r )r r% r )r r rI )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 �t �| ��| � tt �t �| d d d� ��| d d d� � t �| d�} tt jt j| dd�dd�| � | j�� } tt jt j| dd�dd�| � d S )Ni )r r$ )r% r r )r r ) r �aranger r r r� �tiler� r� )�dr r r �test_unpackbits_large s * r� �__main__)� __future__r r r �numpyr Z numpy.testingr r r r r r# r* r� r� r� r� r� r� �__name__r r r r �<module> s !