?????????? ????????? - ??????????????? - /usr/lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_shape_base.cpython-39.pyc
???????
a z[yc�d � @ sz d dl Zd dlZd dlZd dlZd dlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZ d dlmZmZmZmZmZ ejdkZdd� Z G dd� d�Z G d d � d �Z G d d � d �Z G d d� d�Z G dd� d�Z!G dd� d�Z"G dd� d�Z#G dd� d�Z$G dd� d�Z%G dd� d�Z&G dd � d �Z'G d d � d �Z(G d d � d �Z)G d!d"� d"�Z*G d#d$� d$�Z+G d%d&� d&�Z,d'd(� Z-dS ))� N)�apply_along_axis�apply_over_axes� array_split�split�hsplit�dsplit�vsplit�dstack� column_stack�kron�tile� expand_dims�take_along_axis�put_along_axis)�assert_� assert_equal�assert_array_equal� assert_raises� assert_warns� c s t �� �� fdd��}|S )z: hack in keepdims behavior into a function taking an axis c s. � | fd|i|��}|d u r d}t j||d�S )N�axisr �r )�npr )�ar �kwargs�res��func� �E/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_shape_base.py�wrapped s z _add_keepdims.<locals>.wrapped)� functools�wraps)r r r r r � _add_keepdims s r# c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestTakeAlongAxisc C s� ddl m} |ddd�}tjtjt� fttj�ttj�t� fttj �ttj �t� ftj tj tdd�fg}|D ]`\}}}t t|j��dg D ]@}||fd |i|��}||fd |i|��} t|t|| |d �� q�qldS ) z- Test it translates from arg<func> to <func> r ��rand� � � � )ZkthNr r )� numpy.randomr& r �sortZargsort�dictr# �min�argmin�max�argmax� partitionZ argpartition�list�range�ndimr r ) �selfr& r Zfuncsr Zargfuncr r Za_funcZai_funcr r r �test_argequivalent s �z$TestTakeAlongAxis.test_argequivalentc C s� t �d�}t jdt jd�}t||dd� ttt|t �d�dd� ttt||�t �dd� ttt||�t �dd� tt j t||dd� dS )z4 Test it errors when indices has too few dimensions )� r8 )r8 r* ��dtype� r r8 N) r �ones�intpr r � ValueError�array� IndexErrorZastype�bool�float� AxisError�r6 r �air r r � test_invalid1 s z TestTakeAlongAxis.test_invalidc C s: t �d�}t jdt jd�}t||dd�}t|j|j� dS )zC Test everything is ok with empty results, even with inserted dims )r' r( r) )r' r r) r9 r; r N�r r< r= r r �shape�r6 r rE �actualr r r � test_emptyB s z TestTakeAlongAxis.test_emptyc C s8 t �d�}t jdt jd�}t||dd�}t|jd� dS )�D Test that non-indexing dimensions are broadcast in both directions �r' r( r; �r; r* r) r9 r; r �r' r* r) NrG rI r r r �test_broadcastJ s z TestTakeAlongAxis.test_broadcastN)�__name__� __module__� __qualname__r7 rF rK rP r r r r r$ s r$ c @ s e Zd Zdd� Zdd� ZdS )�TestPutAlongAxisc C sz t �g d�g d�g�}tt|j��d g D ]J}|�� }tt j�||d�}t||d|d� tt j �||d�}t ||� q*d S )N)r8 � � )�< �( �2 r i����) r r? r3 r4 r5 �copyr# r1 r r/ r )r6 Za_baser r Zi_maxZi_minr r r �test_replace_maxS s z!TestPutAlongAxis.test_replace_maxc C sL t �d�}t jdt jd��d�d }t||ddd� tt||dd�d� d S ) rL rM r8 r9 rN r( rV r; r N)r r< �aranger= �reshaper r r rD r r r rP c s z TestPutAlongAxis.test_broadcastN)rQ rR rS r[ rP r r r r rT R s rT c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej fd d �Z ej fd d�Z ej fdd�Z dd� Z dd� Zdd� Zdd� Zdd� ZdS ) �TestApplyAlongAxisc C s6 t �dd�}tttd|�t|�t �|jd � � d S )N�rV r8 �dr r; �r r< r r �lenrH �r6 r r r r � test_simplel s �z TestApplyAlongAxis.test_simplec C s6 t �dd�}tttd|�t|�t �|jd � � d S )N)r8 �e r` r r; ra rc r r r �test_simple101q s �z!TestApplyAlongAxis.test_simple101c C s: t �d��d�}ttt jd|�g d�g d�g d�g� d S )N� �r' r' r' r )rg rU �! )�$ �' �* )�- �0 �3 )r r\ r] r r �sumrc r r r �test_3dv s �zTestApplyAlongAxis.test_3dc C s� dd� }G dd� dt j�}t �ddgddgg��|�}t �ddgd d gg��|�}t|d|�}tt||�� t||� t|d|�}tt||�� t||� d S ) Nc S s | d S )Nr* r )�rowr r r �double| s z9TestApplyAlongAxis.test_preserve_subclass.<locals>.doublec @ s e Zd ZdS )z<TestApplyAlongAxis.test_preserve_subclass.<locals>.MyNDArrayN�rQ rR rS r r r r � MyNDArray s ru r r; r* r' r( � )r �ndarrayr? �viewr r � isinstancer )r6 rs ru �m�expected�resultr r r �test_preserve_subclass{ s z)TestApplyAlongAxis.test_preserve_subclassc C sJ G dd� dt j�}dd� }t �d��|�}tt|d|�t �g d��� d S )Nc @ s e Zd ZdZdS )z9TestApplyAlongAxis.test_subclass.<locals>.MinimalSubclassr; N)rQ rR rS �datar r r r �MinimalSubclass� s r c S s | j S �N)r~ )r? r r r �minimal_function� s z:TestApplyAlongAxis.test_subclass.<locals>.minimal_function�rv r' r )r; r; r; )r rw �zerosrx r r r? )r6 r r� r r r r � test_subclass� s �z TestApplyAlongAxis.test_subclassc C sJ t �d��|�}tt jd|�}tt||�� t|t �g d���|�� d S )Nr� r �rv rv rv ) r r< rx r rp r ry r r? )r6 �clsr r r r r �test_scalar_array� s z$TestApplyAlongAxis.test_scalar_arrayc C s� dd� }t �d��|�}t|d|�}tt||�� t|t �g d���|�� t|d|�}tt||�� t|t �g d���|�� d S )Nc S s t | jd� t�tj| dd��S )z/ Sum x, returning a 0d array of the same class r; T)Zkeepdims)r r5 r �squeezerp ��xr r r � sum_to_0d� s z3TestApplyAlongAxis.test_0d_array.<locals>.sum_to_0dr� r r� r; )r' r' r' r' r' r' )r r< rx r r ry r r? )r6 r� r� r r r r r � test_0d_array� 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 )�/produces an asymmetric non-square matrix from xr; N���)r r5 rx r� )r� r r �f1to2� s z5TestApplyAlongAxis.test_axis_insertion.<locals>.f1to2� r� r c s g | ]}�� d d �|f ��qS r� r ��.0�i��a2dr� r r � <listcomp>� s z:TestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>r; r� r c s g | ]}�� |d d �f ��qS r� r r� r� r r r� � s �Z )rv r) r' c s6 g | ].� t j��� fd d�t�jd �D �dd��qS )c s" g | ]}�� |d d ��f ��qS r� r r� )�a3dr� �jr r r� � s zETestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>.<listcomp>r r )r �stackr4 rH )r� )r� r� )r� r r� � s � ��r* ) r r\ r] r r� r4 rH rx r �type)r6 r� rJ r{ r )r� r� r� r� r �test_axis_insertion� sB ��� ��� ���z&TestApplyAlongAxis.test_axis_insertionc C s4 G dd� dt j�}| �|� | �|� | �|� d S )Nc @ s e Zd ZdS )zFTestApplyAlongAxis.test_subclass_preservation.<locals>.MinimalSubclassNrt r r r r r � s r )r rw r� r� r� )r6 r r r r �test_subclass_preservation� 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 )r� r; Nr� r) r )r r5 r �maZ masked_where)r� r r r r r� � s z8TestApplyAlongAxis.test_axis_insertion_ma.<locals>.f1to2r� r� r r' r; r* ) r r\ r] r r ry r� Z masked_arrayr r5 r �mask)r6 r� r r 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 r� r r r � sample_1d� s z7TestApplyAlongAxis.test_tuple_func1d.<locals>.sample_1dr; r* r' r( )r r r? r )r6 r� r 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 r� 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 rb r� r r r � empty_to_1� s z1TestApplyAlongAxis.test_empty.<locals>.empty_to_1)r8 r r8 )r �emptyr r> r r r< )r6 r� r r� rJ r r r rK � s z TestApplyAlongAxis.test_emptyc C s� t �ddhddhddhgddhd d hd d hgg�}t �d d� d|�}t �h d�h d�h d�g�}t||� t �|j�D ] }tt|| �t|| �� qvd S )Nr; � r* � r' ri r( �, r) �7 rv �B c S s t j| � S r� )�set�union)r r r r �<lambda> � z>TestApplyAlongAxis.test_with_iterable_object.<locals>.<lambda>r > r; r� r( r� > r* r) r� r� > ri r� r' rv )r r? r r ZndindexrH r� )r6 r` rJ r{ r� r r r �test_with_iterable_object s � z,TestApplyAlongAxis.test_with_iterable_objectN)rQ rR rS rd rf rq r} r� r rw r� r� r� r� r� r� rK r� r r r r r^ k s %r^ c @ s e Zd Zdd� ZdS )�TestApplyOverAxesc C sH t �d��ddd�}tt j|ddg�}t|t �dgdgdggg�� d S ) N� r* r' r( r rW �\ �| )r r\ r] r rp r r? )r6 r Zaoa_ar r r rd s z TestApplyOverAxes.test_simpleN)rQ rR rS rd r r r r r� s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestExpandDimsc C sR d}t �|�}tdd�D ]4}t||�}t|j| dk� tt �|�j|k� qd S )N�r* r' r( r) ���r( r; )r r� r4 r r rH r� )r6 �sr r �br r r �test_functionality s z!TestExpandDims.test_functionalityc C sn t �d�}t j|dd�jdks"J �t j|dd�jdks:J �t j|dd�jdksRJ �t j|d d�jd ksjJ �d S ) Nrh �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 rH rc r r r �test_axis_tuple! s z TestExpandDims.test_axis_tuplec C s\ d}t �|�}tt jt|d� tt jt|d� t �d�}tt jt|d� tt jt|d� d S )Nr� ���r) rh )r r� )r r) )r r� r rC r )r6 r� r r r r �test_axis_out_of_range( s z%TestExpandDims.test_axis_out_of_rangec C s t �d�}ttt|dd� d S )Nrh �r; r; r )r r� r r> r rc r r r �test_repeated_axis2 s z!TestExpandDims.test_repeated_axisc C sf t �d��d�}t jj||d dkd�}t j|dd�}tt|t jj�� t |j d� t |j j d� d S ) Nr8 )r* r) r' r �r� r; r )r* r; r) ) r r\ r] r� r? r r ry Z MaskedArrayr rH r� )r6 r Zexpandedr r r �test_subclasses6 s z TestExpandDims.test_subclassesN)rQ rR rS r� r� r� r� r� r r r r r� s r� c @ sf e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej j e d d �d d� �Z dd� Z dd� Zdd� ZdS )�TestArraySplitc C s t �d�}ttt|d� d S )Nr8 r )r r\ r r> r rc r r r �test_integer_0_splitA 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 ) Nr8 r; r* r) r' r( � rv � � r� )r r\ r �compare_resultsr? �r6 r r �desiredr r r �test_integer_splitE sr � � � � � � � �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 jju � t|ddgdd�}t �d�t �t �d�g�t �t �d�g�g}t||� t|jj|d jju � d S )Nr8 r' r r �r r8 r� r; � r r? r\ r r� r� r r: r� �r6 r r �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 )Nr8 r' r� r r( r� )r r? r\ r r� 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 jju � dS )z2 This will fail if we change default axis r8 r' r� r� Nr� r� r r r � test_integer_split_2D_default� s � z,TestArraySplit.test_integer_split_2D_defaultzNeeds 64bit platform)�reasonc C s\ t �dgd�}t|d�}t �dgd�}|gd }tt|��D ]}t|| j|| j� q<d S )Nr )r r* r( )i @r* )r Z broadcast_tor r4 rb r rH )r6 r r �chunkr� r� r r r �,test_integer_split_2D_rows_greater_max_int32� s z;TestArraySplit.test_integer_split_2D_rows_greater_max_int32c C sZ t �d�}g d�}t||dd�}t �dd�t �dd�t �dd�t �dd�g}t||� d S ) Nr8 )r; r) r� r� r r r; r) r� )r r\ r r� �r6 r �indicesr r� r r r �test_index_split_simple� s �z&TestArraySplit.test_index_split_simplec C sX t �d�}g d�}t||dd�}t �g �t �dd�t �dd�t �dd�g}t||� d S )Nr8 )r r) r� r� r r r) r� �r r\ r r? r� r� r r r �test_index_split_low_bound� s �z)TestArraySplit.test_index_split_low_boundc C sh t �d�}g d�}t||dd�}t �g �t �dd�t �dd�t �dd�t �g �t �g �g}t||� d S )Nr8 )r r) r� r8 � r� r r r) r� r� r� r r r �test_index_split_high_bound� s �z*TestArraySplit.test_index_split_high_boundN)rQ rR rS r� r� r� r� r� �pytest�markZskipif�IS_64BITr� 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 )Nr8 r* r) )r r\ r r� r� r r r �test_equal_split� s zTestSplit.test_equal_splitc C s t �d�}ttt|d� d S )Nr8 r' )r r\ r r> r rc r r r �test_unequal_split� s z TestSplit.test_unequal_splitN)rQ rR rS r� r� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestColumnStackc C s t ttd� d S �Nr; )r � TypeErrorr �r6 r r r �test_non_iterable� s z!TestColumnStack.test_non_iterablec C sL t �d�}t �d�}t �ddgddgddgg�}t �||f�}t||� d S )N�r; r* r' )r* r' r( r; r* r' r( �r r? r r �r6 r r� r{ rJ r r r �test_1D_arrays� s �z TestColumnStack.test_1D_arraysc C sd t �dgdgdgg�}t �dgdgdgg�}t �ddgddgddgg�}t �||f�}t||� d S )Nr; r* r' r( r� r� r r r �test_2D_arrays� s �z TestColumnStack.test_2D_arraysc C sB t t��& tdd� td�D �� W d � n1 s40 Y d S )Nc s s | ]}t �d �V qdS �r' N�r r\ �r� �_r r r � <genexpr>� r� z1TestColumnStack.test_generator.<locals>.<genexpr>r* )r � FutureWarningr r4 r� r r r �test_generator� s z TestColumnStack.test_generatorN)rQ rR rS r� r� r� r� r r r r r� � s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� TestDstackc C s t ttd� d S r� )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; r* �r r? r r �r6 r r� r r� r r r � test_0D_array� s zTestDstack.test_0D_arrayc C sD t �dg�}t �dg�}t||g�}t �ddggg�}t||� d S r� r� r� r r r � test_1D_array� s zTestDstack.test_1D_arrayc C sX t �dgdgg�}t �dgdgg�}t||g�}t �ddggddggg�}t||� d S r� r� r� r r r � test_2D_array s zTestDstack.test_2D_arrayc C sN t �ddg�}t �ddg�}t||g�}t �ddgddggg�}t||� d S r� r� r� r r r �test_2D_array2 s zTestDstack.test_2D_array2c C sB t t��& tdd� td�D �� W d � n1 s40 Y d S )Nc s s | ]}t �d �V qdS r� r� r� r r r r� r� z,TestDstack.test_generator.<locals>.<genexpr>r* )r r� r r4 r� r r r r� s zTestDstack.test_generatorN) rQ rR rS r� r� r� r� r� r� r r r r r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � TestHsplit�&Only testing for integer splits. c C s t ttdd� d S r� )r r> r r� r r r r� s z TestHsplit.test_non_iterablec C s8 t �d�}zt|d� td� W n ty2 Y n0 d S )Nr; r* r )r r? r r r> rc r r r r� ! s zTestHsplit.test_0D_arrayc C sB t �g d��}t|d�}t �ddg�t �ddg�g}t||� d S �N�r; r* r' r( r* r; r' r( �r r? r r� r� r r r r� ) s zTestHsplit.test_1D_arrayc C sZ t �g d�g d�g�}t|d�}t �ddgddgg�t �ddgddgg�g}t||� d S r� r r� r r r r� / s � ,zTestHsplit.test_2D_arrayN�rQ rR rS �__doc__r� r� r� r� r r r r r� s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � TestVsplitr� c C s t ttdd� d S r� )r r> r r� r r r r� ; s z TestVsplit.test_non_iterablec C s t �d�}ttt|d� d S r� )r r? r r> r rc r r r r� > s zTestVsplit.test_0D_arrayc C s< t �g d��}zt|d� td� W n ty6 Y n0 d S �Nr r* r )r r? r r r> rc r r r r� B s zTestVsplit.test_1D_arrayc C sN t �g d�g d�g�}t|d�}t �g d�g�t �g d�g�g}t||� d S �Nr r* )r r? r r� r� r r r r� J s � zTestVsplit.test_2D_arrayNr r r r r r 7 s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestDsplitc C s t ttdd� d S r� )r r> r r� r r r r� T s z TestDsplit.test_non_iterablec C s t �d�}ttt|d� d S r� �r r? r r> r rc r r r r� W s zTestDsplit.test_0D_arrayc C s t �g d��}ttt|d� d S r r rc r r r r� [ s zTestDsplit.test_1D_arrayc C sD t �g d�g d�g�}zt|d� td� W n ty> Y n0 d S r )r r? r r r> rc r r r r� _ s � zTestDsplit.test_2D_arrayc C s� t �g d�g d�gg d�g d�gg�}t|d�}t �ddgddggddgddggg�t �ddgddggddgddggg�g}t||� d S r� )r r? r r� r� r r r � test_3D_arrayh s ��� $$�zTestDsplit.test_3D_arrayN)rQ rR rS r� 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 r% rV r8 r; )rV r8 r8 )rV r8 rV r_ g �?) r+ r& r r r� r] r r5 r� rw )r6 r& r r� �cr r r r � test_basict s zTestSqueeze.test_basicN�rQ rR rS r r r r r r s s r c @ sT e Zd Zdd� Zej�dejej g�dd� �Z dd� Z ej�dg d ��d d � �Z d S ) �TestKronc C s� G dd� dt j�}t �ddg�}||j|j|j�}ttt||��t j� ttt||��|� ttt||��|� ttt||��|� d S )Nc @ s e Zd ZdZdS )z*TestKron.test_return_type.<locals>.myarrayg �?N)rQ rR rS Z__array_priority__r r r r �myarray� s r r* ) r rw r< rH r: r~ r r� r )r6 r r r� r r r �test_return_type� s zTestKron.test_return_type� array_classc C sL |t �ddg��}|t �ddg��}|t �ddg��}tt �||�|� d S )Nr' r� )r r< r r )r6 r r r� �kr r r �test_kron_smoke� s zTestKron.test_kron_smokec C sj t jjddgddggddgddggd�}t jjt �g d��t jt �d�td� d�}t|t �||�� d S ) Nr; r* r' r( r r� )r; r( r( � r9 )r r� r? Zdiag�identityrA r r )r6 r� r r r r � test_kron_ma� s (�zTestKron.test_kron_mazshape_a,shape_b))r� r� )r� )r( r) rv )�r* r* )r* r* r* ))r; r r� )�r* r r* r ))r* r r r* r c C s� t �|�}t �|�}dtdt|�t|� � | }dtdt|�t|� � | }t �||�}t �||�}t �|j|�s~J d��d S )N)r; r zUnexpected shape from kron)r r< r0 rb Zmultiplyr Z array_equalrH ) r6 Zshape_aZshape_br r� Znormalised_shape_aZnormalised_shape_bZexpected_shaper r r r �test_kron_shape� s ��zTestKron.test_kron_shapeN) rQ rR rS r r� r� Z parametrizer ZasarrayZmatr r r r r r r r � s � � r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestTilec C s� t �g d��}ddgddgg}tt|d�g d�� tt|d�g d�g d�g� tt|d�g d�g� tt|d�g d �g d �g� tt|d �ddgddgddgddgg� tt|d�g d �g d �g d �g d �g� 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* r; )r r? r r �r6 r r� r r r r � s ( �zTestTile.test_basicc C s0 t �d�}t|d�}|d7 }t|t �d�� d S )Nr) r; r* )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 )Nr* rO )r* r )r' r* r )r r? r rH r )r6 r r� r r` r r r rK � s zTestTile.test_emptyc C sn ddl m} g d�}g d�}|D ]H}|dd|d�}|D ]0}t�||j�}t||�}t||�} t|| � q6q d S )Nr )�randint))r* r r r )r* r' r* )r' r* ))r' )r* r' )r' r( r' )r' r* r' )r( r' r* r( r r8 )�size)r+ r r r< r: r r r ) r6 r ZrepsrH r� r� �rr ZlargeZklarger r r �test_kroncompare� s zTestTile.test_kroncompareN)rQ rR rS r r rK r" r r r r r � s r c @ s e Zd Zdd� ZdS )�TestMayShareMemoryc C s� t �d�}t �d�}tt �||�� tt �||d d d� �� tt �||d d d� �� tt �||dd �d d d�f �� tt �|d d d� |� � tt �|d d d� |� � tt �|dd �d d d�f |� � tt �|dd �d d d�f |�� d S )N)rY rW )rU rW rv r� r* r; )r r< r Zmay_share_memory)r6 r` Zd2r r r r � s " $z TestMayShareMemory.test_basicNr r r r r r# � s r# c C s: t | �t |�krtd��t| |�D ]\}}t||� q"dS )zCompare lists of arrays.z Iterables have different lengthsN)rb r> �zipr )r r� r� �yr r r r� � s r� ).Znumpyr r! �sysr� Znumpy.lib.shape_baser r r r r r r r r r r r r r Z numpy.testingr r r r r �maxsizer� r# r$ rT r^ r� r� r� r� r� r� r� r r r r r r# r� r r r r �<module> s4 @ 4 '( ' !2(
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????