Edit File: test_shape_base.cpython-37.pyc
B ��FdL � @ s| d dl mZmZmZ d dlZd dlZd dlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZ d dlmZmZmZmZmZmZmZ G dd� de�Z G dd� de�Z G d d � d e�Z G d d � d e�Z G d d� de�Z G dd� de�Z!G dd� de�Z"G dd� de�Z#G dd� de�Z$G dd� de�Z%G dd� de�Z&G dd � d e�Z'G d d � d e�Z(G d d � d e�Z)d!d"� Z*e+d#k�rxe� dS )$� )�division�absolute_import�print_functionN) �apply_along_axis�apply_over_axes� array_split�split�hsplit�dsplit�vsplit�dstack� column_stack�kron�tile� expand_dims)�run_module_suite�TestCase�assert_� assert_equal�assert_array_equal� assert_raises� assert_warnsc @ s� e Zd Zdd� Zd dd�Zdd� Zdd � Zd d � Zej fd d �Z ej fdd�Z ej fdd�Z dd� Z dd� Zdd� Zdd� Zdd� Zd S ) �TestApplyAlongAxisc C s6 t �dd�}tttd|�t|�t �|jd � � d S )N)� � �dr � )�np�onesr r �len�shape)�self�a� 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_shape_base.py� test_simple s z TestApplyAlongAxis.test_simple� c C s6 t �dd�}tttd|�t|�t �|jd � � d S )N)r �e r r r )r r r r r r )r! �levelr" r# r# r$ �test_simple101 s z!TestApplyAlongAxis.test_simple101c C s@ t �d��d�}ttt jd|�dddgdddgd d d gg� d S ) N� )� r+ r+ r � �! �$ �' �* �- �0 �3 )r �arange�reshaper r �sum)r! r" r# r# r$ �test_3d s zTestApplyAlongAxis.test_3dc C s� dd� }t �ddgddgg�}t �ddgddgg�}t|d|�}tt|t j�� t||� t|d|�}tt|t j�� t||� d S ) Nc S s | d S )N� r# )�rowr# r# r$ �double# s z9TestApplyAlongAxis.test_preserve_subclass.<locals>.doubler r r8 r+ � � )r �matrixr r � isinstancer )r! r: �m�expected�resultr# r# r$ �test_preserve_subclass s z)TestApplyAlongAxis.test_preserve_subclassc C sL G dd� dt j�}dd� }t �d��|�}tt|d|�t �dddg�� d S )Nc @ s e Zd ZdZdS )z9TestApplyAlongAxis.test_subclass.<locals>.MinimalSubclassr N)�__name__� __module__� __qualname__�datar# r# r# r$ �MinimalSubclass1 s rG c S s | j S )N)rF )�arrayr# r# r$ �minimal_function4 s z:TestApplyAlongAxis.test_subclass.<locals>.minimal_function)r< r+ r r )r �ndarray�zeros�viewr r rH )r! rG rI r" r# r# r$ � test_subclass0 s z TestApplyAlongAxis.test_subclassc C sL t �d��|�}tt jd|�}tt||�� t|t �dddg��|�� d S )N)r< r+ r r< ) r r rL r r6 r r> r rH )r! �clsr" �resr# r# r$ �test_scalar_array= s z$TestApplyAlongAxis.test_scalar_arrayc C s� dd� }t �d��|�}t|d|�}tt||�� t|t �dddg��|�� t|d|�}tt||�� t|t �ddddddg��|�� d S )Nc S s t | jd� t�tj| dd��S )z/ Sum x, returning a 0d array of the same class r T)�keepdims)r �ndimr �squeezer6 )�xr# r# r$ � sum_to_0dD s z3TestApplyAlongAxis.test_0d_array.<locals>.sum_to_0d)r< r+ r r< r r+ )r r rL r r r> r rH )r! rN rU r" rO r# r# r$ � test_0d_arrayC s z TestApplyAlongAxis.test_0d_arrayc s2 �fdd��t �d��d�� t�d� �}t j� �fdd�t� jd �D �d d ����}tt |�t |�� t||� t�d� �}t j� �fd d�t� jd �D �dd ����}tt |�t |�� t||� t �d ��d ��t�d��}t j��fdd�t�jd �D �d d ����}tt |�t |�� t||� d S )Nc s0 t | jd� | ddd� | dd�df �� �S )z0produces an assymmetric non-square matrix from xr N���)r rR rL )rT )rN r# r$ �f1to2R s z5TestApplyAlongAxis.test_axis_insertion.<locals>.f1to2� )r< r+ r c s g | ]}�� d d �|f ��qS )Nr# )�.0�i)�a2drX r# r$ � <listcomp>\ s z:TestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>r rW )�axisc s g | ]}�� |d d �f ��qS )Nr# )rZ r[ )r\ rX r# r$ r] d s �Z )r< � r+ c s6 g | ].� t j��� fd d�t�jd �D �dd��qS )c s" g | ]}�� |d d ��f ��qS )Nr# )rZ r[ )�a3drX �jr# r$ r] o s zETestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>.<listcomp>r )r^ )r �stack�ranger )rZ )ra rX )rb r$ r] n s r8 ) r r4 r5 r rc rd r rL r �type)r! rN �actualr@ r# )r\ ra rN rX r$ �test_axis_insertionQ s, z&TestApplyAlongAxis.test_axis_insertionc C s4 G dd� dt j�}| �|� | �|� | �|� d S )Nc @ s e Zd ZdS )zFTestApplyAlongAxis.test_subclass_preservation.<locals>.MinimalSubclassN)rC rD rE r# r# r# r$ rG w s rG )r rJ rP rV rg )r! rG r# r# r$ �test_subclass_preservationv s z-TestApplyAlongAxis.test_subclass_preservationc C s� dd� }t �d��d�}t|d|�}tt|t jj�� t|j d� t |d d �d d �df j ||d d �df �j � t |d d �d d �df j ||d d �df �j � t |d d �d d �df j ||d d �df �j � d S ) Nc S s@ t | jd� | ddd� | dd�df }tj�|d dk|�S )z0produces an assymmetric non-square matrix from xr NrW r` r )r rR r �ma� masked_where)rT rO r# r# r$ rX ~ s z8TestApplyAlongAxis.test_axis_insertion_ma.<locals>.f1to2rY )r< r+ r r+ r r8 ) r r4 r5 r r r> ri � masked_arrayr rR r �mask)r! rX r" rO r# r# r$ �test_axis_insertion_ma} s 00z)TestApplyAlongAxis.test_axis_insertion_mac C sH dd� }t �|dt �ddgddgg��}t|t �ddgddgg�� d S )Nc S s | d | d fS )Nr r r# )rT r# r# r$ � sample_1d� s z7TestApplyAlongAxis.test_tuple_func1d.<locals>.sample_1dr r8 r+ r; )r r rH r )r! rn rO r# r# r$ �test_tuple_func1d� s z$TestApplyAlongAxis.test_tuple_func1dc C s| dd� }t �d�}ttt j|d|� ttt j|d|� dd� }t �d�}t �|d|�}t|t �d �� ttt j|d|� d S ) Nc S s t d� d S )NF)r )rT r# r# r$ � never_call� s z1TestApplyAlongAxis.test_empty.<locals>.never_call)r r r r c S s t t| �dk� dS )Nr r )r r )rT r# r# r$ � empty_to_1� s z1TestApplyAlongAxis.test_empty.<locals>.empty_to_1)r r r )r �emptyr � ValueErrorr r r )r! rp r" rq rf r# r# r$ � test_empty� s z TestApplyAlongAxis.test_emptyc C s� t �tddg�tddg�tddg�gtddg�td d g�td d g�gg�}t �d d� d|�}t �ddddhddd d hddd d hg�}t||� x.t �|j�D ] }tt|| �t|| �� q�W d S )Nr r& r8 � r+ r- r; �, r` �7 r< �B c S s t j| � S )N)�set�union)r" r# r# r$ �<lambda>� � z>TestApplyAlongAxis.test_with_iterable_object.<locals>.<lambda>r )r rH ry r r �ndindexr re )r! r rf r@ r[ r# r# r$ �test_with_iterable_object� s &( z,TestApplyAlongAxis.test_with_iterable_objectN)r& )rC rD rE r% r) r7 rB rM r rJ rP rV rg rh rm ro rt r~ r# r# r# r$ r s %r c @ s e Zd Zdd� ZdS )�TestApplyOverAxesc C sH t �d��ddd�}tt j|ddg�}t|t �dgdgdggg�� d S ) N� r8 r+ r; r �< �\ �| )r r4 r5 r r6 r rH )r! r" Zaoa_ar# r# r$ r% � s z TestApplyOverAxes.test_simpleN)rC rD rE r% r# r# r# r$ r � s r c @ s e Zd Zdd� Zdd� ZdS )�TestExpandDimsc C sV d}t �|�}xBtdd�D ]4}t||�}t|j| dk� tt �|�j|k� qW d S )N)r8 r+ r; r` ���r; r )r rr rd r r r rS )r! �sr" r^ �br# r# r$ �test_functionality� s z!TestExpandDims.test_functionalityc C sL d}t �|�}t�� �, t�d� ttt|d� ttt|d� W d Q R X d S )N)r8 r+ r; r` �alwaysi����r` )r rr �warnings�catch_warnings� simplefilterr �DeprecationWarningr )r! r� r" r# r# r$ �test_deprecations� s z TestExpandDims.test_deprecationsN)rC rD rE r� r� r# r# r# r$ r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestArraySplitc C s t �d�}ttt|d� d S )Nr r )r r4 r rs r )r! r" r# r# r$ �test_integer_0_split� s z#TestArraySplit.test_integer_0_splitc C s� t �d�}t|d�}t �d�g}t||� t|d�}t �d�t �dd�g}t||� t|d�}t �d�t �dd�t �dd�g}t||� t|d�}t �d�t �dd�t �dd �t �d d�g}t||� t|d�}t �d�t �dd�t �dd�t �dd �t �d d�g}t||� t|d�}t �d�t �dd�t �dd�t �dd �t �d d �t �d d�g}t||� t|d�}t �d�t �dd�t �dd�t �dd�t �dd �t �d d �t �d d�g}t||� t|d �}t �d�t �dd�t �dd�t �dd�t �dd�t �dd �t �d d �t �d d�g}t||� t|d �}t �d�t �dd�t �dd�t �dd�t �dd�t �dd�t �dd �t �d d �t �d d�g }t||� t|d�}t �d�t �dd�t �dd�t �dd�t �dd�t �dd�t �dd�t �dd �t �d d �t �d d�g }t||� t|d �}t �d�t �dd�t �dd�t �dd�t �dd�t �dd�t �dd�t �dd �t �d d �t �d d�t �g �g }t||� d S ) Nr r r8 r` r+ r; � r< � � r& )r r4 r �compare_resultsrH )r! r" rO �desiredr# r# r$ �test_integer_split� sb " " z!TestArraySplit.test_integer_splitc C s� t �t �d�t �d�g�}t|ddd�}t �t �d�g�t �t �d�g�t �d�g}t||� t|jj|d jjk� t|dddgdd�}t �d�t �t �d�g�t �t �d�g�g}t||� t|jj|d jjk� d S ) Nr r+ r )r^ )r r rW r r8 ) r rH r4 r rK r� r �dtypere )r! r" rO �tgtr# r# r$ �test_integer_split_2D_rows s z)TestArraySplit.test_integer_split_2D_rowsc C s� t �t �d�t �d�g�}t|ddd�}t �t �d�t �d�g�t �t �dd�t �dd�g�t �t �dd�t �dd�g�g}t||� d S )Nr r+ rW )r^ r; r� )r rH r4 r r� )r! r" rO r� r# r# r$ �test_integer_split_2D_cols s z)TestArraySplit.test_integer_split_2D_colsc C sv t �t �d�t �d�g�}t|d�}t �t �d�g�t �t �d�g�t �d�g}t||� t|jj|d jjk� dS )z2 This will fail if we change default axis r r+ )r r rW N) r rH r4 r rK r� r r� re )r! r" rO r� r# r# r$ � test_integer_split_2D_default' s z,TestArraySplit.test_integer_split_2D_defaultc C s\ t �d�}dddg}t||dd�}t �dd�t �dd�t �dd�t �dd�g}t||� d S )Nr r r` r� rW )r^ r )r r4 r r� )r! r" �indicesrO r� r# r# r$ �test_index_split_simple2 s z&TestArraySplit.test_index_split_simplec C sZ t �d�}dddg}t||dd�}t �g �t �dd�t �dd�t �dd�g}t||� d S )Nr r r` r� rW )r^ )r r4 r rH r� )r! r" r� rO r� r# r# r$ �test_index_split_low_bound: s z)TestArraySplit.test_index_split_low_boundc C sn t �d�}dddddg}t||dd�}t �g �t �dd�t �dd�t �dd�t �g �t �g �g}t||� d S )Nr r r` r� � rW )r^ )r r4 r rH r� )r! r" r� rO r� r# r# r$ �test_index_split_high_boundB s z*TestArraySplit.test_index_split_high_boundN) rC rD rE r� r� r� r� r� r� r� r� r# r# r# r$ r� � s = r� c @ s e Zd Zdd� Zdd� ZdS )� TestSplitc C s8 t �d�}t|d�}t �d�t �dd�g}t||� d S )Nr r8 r` )r r4 r r� )r! r" rO r� r# r# r$ �test_equal_splitP s zTestSplit.test_equal_splitc C s t �d�}ttt|d� d S )Nr r+ )r r4 r rs r )r! r" r# r# r$ �test_unequal_splitV s z TestSplit.test_unequal_splitN)rC rD rE r� r� r# r# r# r$ r� K s r� c @ s e Zd Zdd� ZdS )�TestColumnStackc C s t ttd� d S )Nr )r � TypeErrorr )r! r# r# r$ �test_non_iterable[ s z!TestColumnStack.test_non_iterableN)rC rD rE r� r# r# r# r$ r� Z s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestDstackc C s t ttd� d S )Nr )r r� r )r! r# r# r$ r� ` s z TestDstack.test_non_iterablec C s@ t �d�}t �d�}t||g�}t �ddggg�}t||� d S )Nr r8 )r rH r r )r! r" r� rO r� r# r# r$ � test_0D_arrayc s zTestDstack.test_0D_arrayc C sD t �dg�}t �dg�}t||g�}t �ddggg�}t||� d S )Nr r8 )r rH r r )r! r" r� rO r� r# r# r$ � test_1D_arrayj s zTestDstack.test_1D_arrayc C sX t �dgdgg�}t �dgdgg�}t||g�}t �ddggddggg�}t||� d S )Nr r8 )r rH r r )r! r" r� rO r� r# r# r$ � test_2D_arrayq s zTestDstack.test_2D_arrayc C sN t �ddg�}t �ddg�}t||g�}t �ddgddggg�}t||� d S )Nr r8 )r rH r r )r! r" r� rO r� r# r# r$ �test_2D_array2x s zTestDstack.test_2D_array2N)rC rD rE r� r� r� r� r� r# r# r# r$ r� _ s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � TestHsplitz&Only testing for integer splits. c C s t ttdd� d S )Nr )r rs r )r! r# r# r$ r� � s z TestHsplit.test_non_iterablec C s: t �d�}yt|d� td� W n tk r4 Y nX d S )Nr r8 r )r rH r r rs )r! r" r# r# r$ r� � s zTestHsplit.test_0D_arrayc C sF t �ddddg�}t|d�}t �ddg�t �ddg�g}t||� d S )Nr r8 r+ r; )r rH r r� )r! r" rO r� r# r# r$ r� � s zTestHsplit.test_1D_arrayc C sb t �ddddgddddgg�}t|d�}t �ddgddgg�t �ddgddgg�g}t||� d S )Nr r8 r+ r; )r rH r r� )r! r" rO r� r# r# r$ r� � s ,zTestHsplit.test_2D_arrayN)rC rD rE �__doc__r� r� r� r� r# r# r# r$ r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � TestVsplitz&Only testing for integer splits. c C s t ttdd� d S )Nr )r rs r )r! r# r# r$ r� � s z TestVsplit.test_non_iterablec C s t �d�}ttt|d� d S )Nr r8 )r rH r rs r )r! r" r# r# r$ r� � s zTestVsplit.test_0D_arrayc C sB t �ddddg�}yt|d� td� W n tk r< Y nX d S )Nr r8 r+ r; r )r rH r r rs )r! r" r# r# r$ r� � s zTestVsplit.test_1D_arrayc C s^ t �ddddgddddgg�}t|d�}t �ddddgg�t �ddddgg�g}t||� d S )Nr r8 r+ r; )r rH r r� )r! r" rO r� r# r# r$ r� � s (zTestVsplit.test_2D_arrayN)rC rD rE r� r� r� r� r� r# r# r# r$ r� � s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestDsplitc C s t ttdd� d S )Nr )r rs r )r! r# r# r$ r� � s z TestDsplit.test_non_iterablec C s t �d�}ttt|d� d S )Nr r8 )r rH r rs r )r! r" r# r# r$ r� � s zTestDsplit.test_0D_arrayc C s$ t �ddddg�}ttt|d� d S )Nr r8 r+ r; )r rH r rs r )r! r" r# r# r$ r� � s zTestDsplit.test_1D_arrayc C sN t �ddddgddddgg�}yt|d� td� W n tk rH Y nX d S )Nr r8 r+ r; r )r rH r r rs )r! r" r# r# r$ r� � s zTestDsplit.test_2D_arrayc C s� t �ddddgddddggddddgddddggg�}t|d�}t �ddgddggddgddggg�t �ddgddggddgddggg�g}t||� d S )Nr r8 r+ r; )r rH r r� )r! r" rO r� r# r# r$ � test_3D_array� s $(zTestDsplit.test_3D_arrayN)rC rD rE r� r� r� r� r� r# r# r# r$ r� � s r� c @ s e Zd Zdd� ZdS )� TestSqueezec C s� ddl m} |ddddd�}|ddddd�}|dddd�}tt�|�t�|d�� tt�|�t�|d�� tt�|�t�|d�� d ggg}t�|�}t|d � t|jd� tt|�tj � d S ) Nr )�randr r r )r r r )r r r )r r g �?) � numpy.randomr� r r rS r5 r rR re rJ )r! r� r" r� �crO r# r# r$ � test_basic� s zTestSqueeze.test_basicN)rC rD rE r� r# r# r# r$ r� � s r� c @ s e Zd Zdd� ZdS )�TestKronc C s� t �ddg�}t �|�}ttt||��t j� ttt||��t j� ttt||��t j� ttt||��t j� G dd� dt j�}||j|j |j �}ttt||��t j� ttt||��|� ttt||��t j� ttt||��|� d S )Nr8 c @ s e Zd ZdZdS )z*TestKron.test_return_type.<locals>.myarrayg N)rC rD rE �__array_priority__r# r# r# r$ �myarray� s r� ) r r �asmatrixr re r rJ r= r r� rF )r! r" r? r� ri r# r# r$ �test_return_type� s zTestKron.test_return_typeN)rC rD rE r� r# r# r# r$ r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestTilec C s t �dddg�}ddgddgg}tt|d�ddddddg� tt|d�ddddddgddddddgg� tt|d�ddddddgg� tt|d�ddddgddddgg� tt|d�ddgddgddgddgg� tt|d�ddddgddddgddddgddddgg� d S ) Nr r r8 r+ r; )r8 r8 )r r8 )r8 r )r rH r r )r! r" r� r# r# r$ r� s , $( zTestTile.test_basicc C s0 t �d�}t|d�}|d7 }t|t �d�� d S )Nr` r r8 )r r4 r r )r! r" r� r# r# r$ �(test_tile_one_repetition_on_array_gh4679 s z1TestTile.test_tile_one_repetition_on_array_gh4679c C sL t �g gg�}t �g g g�}t|d�j}t|d�j}t|d� t|d� d S )Nr8 )r+ r8 r` )r8 r )r+ r8 r )r rH r r r )r! r" r� r� r r# r# r$ rt s zTestTile.test_emptyc C s� ddl m} ddddddg}d d d d d dg}xT|D ]L}|dd|d�}x8|D ]0}t�||j�}t||�}t||�} t|| � qJW q2W d S )Nr )�randint)r8 )r r8 )r8 r )r8 r8 )r8 r+ r8 )r+ r8 )r+ )r8 r+ )r+ r; r+ )r+ r8 r+ )r; r+ r8 r; r )�size)r� r� r r r� r r r ) r! r� �repsr r� r� �rr" ZlargeZklarger# r# r$ �test_kroncompare s zTestTile.test_kroncompareN)rC rD rE r� r� rt r� r# r# r# r$ r� s r� c @ s e Zd Zdd� ZdS )�TestMayShareMemoryc C s t �d�}t �d�}| �t �||�� | �t �||d d d� �� | �t �||d d d� �� | �t �||dd �d d d�f �� | �t �|d d d� |�� | �t �|d d d� |�� | �t �|dd �d d d�f |�� | �t �|dd �d d d�f |�� d S )N)�2 r� )r, r� r< rW r8 r )r r � assertTrue�may_share_memory� assertFalse)r! r �d2r# r# r$ r� * s $ $z TestMayShareMemory.test_basicN)rC rD rE r� r# r# r# r$ r� ) s r� c C s, x&t t|��D ]}t| | || � qW d S )N)rd r r )rO r� r[ r# r# r$ r� 9 s r� �__main__),� __future__r r r �numpyr r� Znumpy.lib.shape_baser r r r r r r r r r r r Z numpy.testingr r r r r r r r r r� r� r� r� r� r� r� r� r� r� r� r� r� rC r# r# r# r$ �<module> s, 8$ %}# !(