a; D d dl Zd dlZd dlZd dlZd dlmZmZmZm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZ d dlmZmZmZmZmZ ej dk Zd Z G d d Z G d d Z G d
d Z G d d
Z G d d Z! G d d Z" G d d Z# G d d Z$ G d d Z% G d d Z& G d d Z' G d d Z( G d d Z) G d d! Z* G d" d# Z+ G d$ d% Z,d& Z-dS )' N)apply_along_axisapply_over_axes array_splitsplithsplitdsplitvsplitdstack column_stackkrontile expand_dimstake_along_axisput_along_axis)assert_ assert_equalassert_array_equal
assert_raises assert_warns c F t j fd }|S )z: hack in keepdims behavior into a function taking an axis c N | fd|i|}|d}t j || S )Naxisr r )npr )ar kwargsresfuncs s/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test_shape_base.pywrappedz _add_keepdims..wrapped s> d1**4*6** <D~c---- ) functoolswraps)r r! s ` r
_add_keepdimsr% s8 _T. . . . .
Nr" c & e Zd Zd Zd Zd Zd ZdS )TestTakeAlongAxisc
ddl m} |ddd }t j t j t ft
t j t
t j t ft
t j t
t j
t ft j t j t d fg}|D ]d\ }}}t t |j dgz D ]8} ||fd |i|} ||fd |i|} t! |t# || |
9edS ) z- Test it translates from arg to r rand )kthNr r ) numpy.randomr* r sortargsortdictr% minargminmaxargmax partition argpartitionlistrangendimr r )
selfr* r funcsr argfuncr r a_funcai_funcs
r test_argequivalentz$TestTakeAlongAxis.test_argequivalent sH %%%%%% DAqMM Wbj$&& )
26
"
"M")$<$.double| s 7Nr" c e Zd ZdS ).MyNDArrayNr` ra rb rc r" r MyNDArrayr Dr" r r rG r. r+ r, )r ndarrayrK viewr r
isinstancer )r= r r mexpectedresults r test_preserve_subclassz)TestApplyAlongAxis.test_preserve_subclass{ s
Hq!fq!f% & & + +I 6 68aVaV ,--229==!&!Q//
69--...68,,,!&!Q//
69--...68,,,,,r" c G d dt j }d }t j d | }t t |d| t j g d d S )Nc e Zd ZdZdS )9TestApplyAlongAxis.test_subclass..MinimalSubclassrG N)r` ra rb datarc r" r MinimalSubclassr s DDDr" r c | j S N)r )rK s r minimal_functionz:TestApplyAlongAxis.test_subclass..minimal_function s
: r" r r+ r )rG rG rG )r r zerosr r r rK )r= r r r s r
test_subclassz TestApplyAlongAxis.test_subclass s bj
HV ! !/ 2 2 -q! 4 4bhyyy6I6I
r" c 2 t j d | }t t j d| }t t
|| t |t j g d | d S )Nr r r r r ) r rH r r r r r r rK )r= clsr r s r test_scalar_arrayz$TestApplyAlongAxis.test_scalar_array sz GFOO % % rvq!,,
3$$%%%3 3 3 8 8 = =>>>>>r" c d }t j d | }t |d| }t t || t
|t j g d | t |d| }t t || t
|t j g d | d S )Nc | t | j d t j t j | d S )z/ Sum x, returning a 0d array of the same class rG T)keepdims)r r< r squeezer xs r sum_to_0dz3TestApplyAlongAxis.test_0d_array..sum_to_0d s3 # # #:bfQ 6 6 677 7r" r r r rG )r+ r+ r+ r+ r+ r+ )r rH r r r r r rK )r= r r r r s r
test_0d_arrayz TestApplyAlongAxis.test_0d_array s 8 8 8
GFOO % % y!Q//
3$$%%%3 3 3 8 8 = =>>> y!Q//
3$$%%%3););); < < A A# F FGGGGGr" c J fdt j d d t d }t j fdt j d D d }t t | t | t || t d }t j fd t j d D d }t t | t | t || t j d
d t d }t j fd t j d
D d }t t | t | t || d S )Nc t | j d | ddd | dddf z S )/produces an asymmetric non-square matrix from xrG N)r r< r )r r s r f1to2z5TestApplyAlongAxis.test_axis_insertion..f1to2 sD # # #dddGa4j(..s33 3r" r r c : g | ]} d d |f S r rc .0ia2dr s r
z:TestApplyAlongAxis.test_axis_insertion.. s:
!EE#aaac(OO
r" rG r r c : g | ]} |d d f S r rc r s r r z:TestApplyAlongAxis.test_axis_insertion.. s:
!EE#ac(OO
r" Z )r r- r+ c g | ]>t j fd t j d D d ?S )c < g | ]} |d d f S r rc )r r a3dr js r r zETestApplyAlongAxis.test_axis_insertion... s> &'c!AAAa%j!! r" r r )r stackr; rV )r r r r s @r r z:TestApplyAlongAxis.test_axis_insertion.. s~
H +01 +>+>
r" r. )
r rr rs r r r; rV r r type)r= r rX r r r r s ` @@@r test_axis_insertionz&TestApplyAlongAxis.test_axis_insertion sH 4 4 4 4 4
inn$$V,, "%C008
%*39Q<%8%8
DII T&\\4>>222VX&&& "%C008
%*39Q<%8%8
499 T&\\4>>222VX&&& i &&y11!%C008
39Q<((
DII T&\\4>>222VX&&&&&r" c G d dt j }| | | | | | d S )Nc e Zd ZdS )FTestApplyAlongAxis.test_subclass_preservation..MinimalSubclassNr rc r" r r r r r" r )r r r r r )r= r s r test_subclass_preservationz-TestApplyAlongAxis.test_subclass_preservation sp bj /// ?+++ 11111r" c Z d }t j d d }t |d| }t t |t j j 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 t | j d | ddd | dddf z }t j |dz dk | S )r rG Nr r- r )r r< r ma masked_where)r r s r r z8TestApplyAlongAxis.test_axis_insertion_ma..f1to2 sT # # #DDbD'AabbgJ&C5%%c!eQh44 4r" r r r r+ rG r. ) r rr rs r r r r masked_arrayr r< r mask)r= r r r s r test_axis_insertion_maz)TestApplyAlongAxis.test_axis_insertion_ma s% 5 5 5
IcNN " "6 * * ua++
3 233444SXq!!!3qqq1u:?EE!AAAaC&MM,>???3qqq1u:?EE!AAAaC&MM,>???3qqq1u:?EE!AAAaC&MM,>?????r" c d }t j |dt j ddgddgg }t |t j ddgddgg d S )Nc " | d | d fS )NrG r rc r s r sample_1dz7TestApplyAlongAxis.test_tuple_func1d..sample_1d s Q41: r" rG r. r+ r, )r r rK r )r= r r s r test_tuple_func1dz$TestApplyAlongAxis.test_tuple_func1d si !)Q1a&1a&9I0J0JKK31a&1a&)9 : :;;;;;r" c d }t j d }t t t j |d| t t t j |d| d }t j d }t j |d| }t |t j d t t t j |d| d S )Nc $ t d d S )NF)r r s r
never_callz1TestApplyAlongAxis.test_empty..never_call s ENNNNNr" )r r r rG c F t t | dk dS )Nr rG )r rz r s r
empty_to_1z1TestApplyAlongAxis.test_empty..empty_to_1 s CFFaK 1r" )rD r rD )r emptyr rJ r r rH )r= r r r rX s r rY z TestApplyAlongAxis.test_empty s
HV j""5z1aHHHj""5z1aHHH
HW $ZA66VRWR[[)))j""5z1aHHHHHr" c t j ddhddhddhgddhd d
hd d hgg }t j d
d| }t j h dh dh dg }t || t j |j D ]8}t t
|| t
|| 9d S )NrG r. r+ r r, , r- 7 r B c t j | S r )setunion)r s r z>TestApplyAlongAxis.test_with_iterable_object.. s sy!} r" r > rG r, r r > r. r- r r > r+ r r r )r rK r r ndindexrV r )r= rx rX r r s r test_with_iterable_objectz,TestApplyAlongAxis.test_with_iterable_object s HWq"g2w 'Wq"g2w '
$%<%>>>>>r" c t j d d }t j ||dz dk }t j |d }t
t |t j j t |j
d t |j j
d d S ) NrD )r. r- r+ r r rG r )r. rG r- ) r rr rs r rK r r r MaskedArrayr rV r )r= r expandeds r test_subclassesz TestExpandDims.test_subclasses6 s IbMM ! !& ) ) EKK!qK ) )>!!,,,
8RU%677888X^Y///X]()44444r" N)r` ra rb r r r r r rc r" r r r sb . . .O O O< < <? ? ?5 5 5 5 5r" r c e Zd Zd Zd Zd Zd Zd Zej
e d d Z d Z
d
Zd Zd S )
TestArraySplitc f t j d }t t t |d d S )NrD r )r rr r rJ r r{ s r test_integer_0_splitz#TestArraySplit.test_integer_0_splitA s) IbMMj+q!44444r" c
R
t j d }t |d }t j d g}t || t |d }t j d t j dd g}t || t |d }t j d t j dd t j dd g}t || t |d }t j d t j dd t j dd t j d d g}t || t |d }t j d t j dd t j dd t j dd t j d d g}t || t |d }t j d t j dd t j dd t j dd t j d d
t j d
d g}t || t |d }t j d t j dd t j dd t j dd t j dd t j d d
t j d
d g}t || t |d }t j d t j dd t j dd t j dd t j dd t j dd t j d d
t j d
d g}t || t |d
}t j d t j dd t j dd t j dd t j dd t j dd t j dd t j d d
t j d
d g }t || t |d }t j d t j dd t j dd t j dd t j dd t j dd t j dd t j dd t j d d
t j d
d g
}t || t |d }t j d t j dd t j dd t j dd t j dd t j dd t j dd t j dd t j d d
t j d
d t j g g }t || d S ) NrD rG r. r- r+ r, r r )r rr r compare_resultsrK r= r r desireds r test_integer_splitz!TestArraySplit.test_integer_splitE s IbMM!Q 9R==/W%%%!Q 9Q<<1b!1!12W%%%!Q 9Q<<1a ")Ar2B2BCW%%%!Q 9Q<<1a ")Aq//9Q##%W%%%!Q 9Q<<1a ")Aq//9Q??BIa$4$46W%%%!Q 9Q<<1a ")Aq//9Q??BIaOORYq"5E5EGW%%%!Q 9Q<<1a ")Aq//9Q??BIaOORYq!__9Q##% W%%%!Q 9Q<<1a ")Aq//9Q??BIaOORYq!__9Q??BIa$4$46 W%%%!Q 9Q<<1a ")Aq//9Q??BIaOORYq!__9Q??BIaOORYq"5E5EG W%%%!R 9Q<<1a ")Aq//9Q??BIaOORYq!__9Q??BIaOORYq!__9Q##% W%%%!R 9Q<<1a ")Aq//9Q??BIaOORYq!__9Q??BIaOORYq!__9Q##RXb\\3 W%%%%%r" c 6 t j t j d t j d g }t |dd }t j t j d g t j t j d g t j d g}t || t
|j j |d j j u t |ddgd }t j d t j t j d g t j t j d g g}t || t
|j j |d j j u d S )NrD r+ r r r rD r rG r rK rr r r r r rF r r= r r tgts r test_integer_split_2D_rowsz)TestArraySplit.test_integer_split_2D_rows s< HbimmRYr]]3 4 4!QQ'''x2 (("(BIbMM?*C*C8G$$&S!!! B
22333 !aV!,,,x "(BIbMM?";";x2 ((*S!!! B
2233333r" c t j t j d t j d g }t |dd }t j t j d t j d g t j t j dd t j dd g t j t j dd t j dd g g}t || d S )NrD r+ r r r, r )r rK rr r r r s r test_integer_split_2D_colsz)TestArraySplit.test_integer_split_2D_cols s HbimmRYr]]3 4 4!QR(((8RYq\\29Q<< 8998RYq!__bi1oo >??8RYq" - -ryB/?/? @AAC W%%%%%r" c t j t j d t j d g }t |d }t j t j d g t j t j d g t j d g}t || t
|j j |d j j u dS )z2 This will fail if we change default axis
rD r+ r r Nr r s r test_integer_split_2D_defaultz,TestArraySplit.test_integer_split_2D_default s
HbimmRYr]]3 4 4!Q x2 (("(BIbMM?*C*C8G$$&S!!! B
2233333r" zNeeds 64bit platform)reasonc t j dgd }t |d }t j dgd }|gdz }t t | D ](}t || j || j )d S )Nr )r r. r, )i @r. )r broadcast_tor r; rz r rV )r= r r chunkr r s r ,test_integer_split_2D_rows_greater_max_int32z;TestArraySplit.test_integer_split_2D_rows_greater_max_int32 s OQC . .!Q \22gks3xx 5 5A Q s1v| 4 4 4 4 5 5r" c t j d }g d}t ||d }t j dd t j dd t j dd t j dd g}t || d S ) NrD )rG r- r r r r rG r- r )r rr r r r= r indicesr r s r test_index_split_simplez&TestArraySplit.test_index_split_simple s} IbMM))!W2...9Q??BIaOORYq!__9Q##%W%%%%%r" c t j d }g d}t ||d }t j g t j dd t j dd t j dd g}t || d S )NrD )r r- r r r r r- r r rr r rK r r% s r test_index_split_low_boundz)TestArraySplit.test_index_split_low_bound s{ IbMM))!W2...8B<<1a ")Aq//9Q##%W%%%%%r" c h t j d }g d}t ||d }t j g t j dd t j dd t j dd t j g t j g g}t || d S )NrD )r r- r rD r r r r- r r) r% s r test_index_split_high_boundz*TestArraySplit.test_index_split_high_bound s IbMM###!W2...8B<<1a ")Aq//9Q##RXb\\28B<<AW%%%%%r" N)r` ra rb r r r r r pytestmarkskipifIS_64BITr# r' r* r- rc r" r r
r
@ s 5 5 5;& ;& ;&z
4
4
4 & & &4 4 4 [H -CDD5 5 ED5& & && & && & & & &r" r
c e Zd Zd Zd ZdS ) TestSplitc t j d }t |d }t j d t j dd g}t || d S )NrD r. r- )r rr r r r s r test_equal_splitzTestSplit.test_equal_split sP IbMMAqkk9Q<<1b!1!12W%%%%%r" c f t j d }t t t |d d S )NrD r+ )r rr r rJ r r{ s r test_unequal_splitz TestSplit.test_unequal_split s) IbMMj%A.....r" N)r` ra rb r5 r7 rc r" r r3 r3 s2
& & & / / / / /r" r3 c & e Zd Zd Zd Zd Zd ZdS )TestColumnStackc < t t t d d S NrG )r TypeErrorr r= s r test_non_iterablez!TestColumnStack.test_non_iterable s i q11111r" c t j d }t j d }t j ddgddgddgg }t j ||f }t || d S )NrG r. r+ )r. r+ r, rG r. r+ r, r rK r r r= r r r rX s r test_1D_arraysz TestColumnStack.test_1D_arrays su HY HY 8aV V V % & & !Q((VX&&&&&r" c t j dgdgdgg }t j dgdgdgg }t j ddgddgddgg }t j ||f }t || d S )NrG r. r+ r, rA rB s r test_2D_arraysz TestColumnStack.test_2D_arrays s HqcA3_ % % HqcA3_ % %8aV V V % & & !Q((VX&&&&&r" c t j t d 5 t d t d D d d d d S # 1 swxY w Y d S )Narrays to stack must bematchc 3 > K | ]}t j d V dS r+ Nr rr r _s r z1TestColumnStack.test_generator.. s* 991")A,,999999r" r. )r. raisesr< r r; r= s r test_generatorz TestColumnStack.test_generator s
]9,E
F
F
F ; ; 99a999 : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; 'AAAN)r` ra rb r> rC rE rQ rc r" r r9 r9 sP 2 2 2' ' '' ' '; ; ; ; ;r" r9 c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )
TestDstackc < t t t d d S r; )r r< r
r= s r r> z TestDstack.test_non_iterable s i+++++r" c t j d }t j d }t ||g }t j ddggg }t || d S NrG r. r rK r
r r= r r r r s r
test_0D_arrayzTestDstack.test_0D_array sX HQKK HQKKaVnn(aVH:&&3(((((r" c t j dg }t j dg }t ||g }t j ddggg }t || d S rW rX rY s r
test_1D_arrayzTestDstack.test_1D_array s\ HaSMM HaSMMaVnn(aVH:&&3(((((r" c t j dgdgg }t j dgdgg }t ||g }t j ddggddggg }t || d S rW rX rY s r
test_2D_arrayzTestDstack.test_2D_array su HqcA3Z HqcA3Z aVnn(aVH1xj1223(((((r" c t j ddg }t j ddg }t ||g }t j ddgddggg }t || d S rW rX rY s r test_2D_array2zTestDstack.test_2D_array2 sl HaV HaV aVnn(aVaV ,-..3(((((r" c t j t d 5 t d t d D d d d d S # 1 swxY w Y d S )NrG rH c 3 > K | ]}t j d V dS rK rL rM s r rO z,TestDstack.test_generator.. s* 33QBIaLL333333r" r. )r. rP r< r
r; r= s r rQ zTestDstack.test_generator s
]9,E
F
F
F 5 5 33%((333 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5rR N) r` ra rb r> rZ r\ r^ r` rQ rc r" r rT rT sn , , ,) ) )) ) )) ) )) ) )5 5 5 5 5r" rT c * e Zd ZdZd Zd Zd Zd ZdS )
TestHsplit&Only testing for integer splits.
c > t t t dd d S r; )r rJ r r= s r r> z TestHsplit.test_non_iterable j&!Q/////r" c t j d } t |d t d d S # t $ r Y d S w xY w)NrG r. r )r rK r r rJ r{ s r rZ zTestHsplit.test_0D_array! sR HQKK 1aLLL AJJJJJ DD s 7
AAc t j g d }t |d }t j ddg t j ddg g}t || d S NrG r. r+ r, r. rG r+ r, r rK r r r s r r\ zTestHsplit.test_1D_array) s^ H\\\ " "Qll8QF##RXq!f%5%56W%%%%%r" c t j g dg dg }t |d }t j ddgddgg t j ddgddgg g}t || d S rj rl r s r r^ zTestHsplit.test_2D_array/ s Hlll ,,
!
!Qll8aVaV ,--rx!Q!Q8H/I/IJW%%%%%r" Nr` ra rb __doc__r> rZ r\ r^ rc r" r rd rd sZ 0 0 0 & & & & & & & &r" rd c * e Zd ZdZd Zd Zd Zd ZdS )
TestVsplitre c > t t t dd d S r; )r rJ r r= s r r> z TestVsplit.test_non_iterable; rg r" c f t j d }t t t |d d S rW )r rK r rJ r r{ s r rZ zTestVsplit.test_0D_array> ) HQKKj&!Q/////r" c t j g d } t |d t d d S # t $ r Y d S w xY wNrk r. r )r rK r r rJ r{ s r r\ zTestVsplit.test_1D_arrayB sX H\\\ " " 1aLLL AJJJJJ DD s 9
AAc t j g dg dg }t |d }t j g dg t j g dg g}t || d S Nrk r. )r rK r r r s r r^ zTestVsplit.test_2D_arrayJ sr Hlll ,,
!
!Qll8\\\N++RX|||n-E-EFW%%%%%r" Nrn rc r" r rq rq 7 sZ 0 0 00 0 0 & & & & &r" rq c , e Zd Zd Zd Zd Zd Zd ZdS )
TestDsplitc > t t t dd d S r; )r rJ r r= s r r> z TestDsplit.test_non_iterableT rg r" c f t j d }t t t |d d S rW r rK r rJ r r{ s r rZ zTestDsplit.test_0D_arrayW rt r" c j t j g d }t t t |d d S rx r} r{ s r r\ zTestDsplit.test_1D_array[ s/ H\\\ " "j&!Q/////r" c t j g dg dg } t |d t d d S # t $ r Y d S w xY wrv )r rK r r rJ r{ s r r^ zTestDsplit.test_2D_array_ sh Hlll ,,
!
! 1aLLL AJJJJJ DD s =
A
A c t j g dg dgg dg dgg }t |d }t j ddgddggddgddggg t j ddgddggddgddggg g}t || d S rj )r rK r r r s r
test_3D_arrayzTestDsplit.test_3D_arrayh s H||| <<! << <<!"
#
# Qll8q!fq!f -AA/? @AA8q!fq!f -AA/? @AACW%%%%%r" N)r` ra rb r> rZ r\ r^ r rc r" r rz rz R s_ 0 0 00 0 00 0 0 & & & & &r" rz c e Zd Zd ZdS ) TestSqueezec r ddl m} |ddddd } |ddddd } |dddd }t t j | t j |d t t j | t j |d t t j | t j |d d ggg}t j | }t
|d t
|j d t
t | t j d S )
Nr r) rh rD rG )rh rD rD )rh rD rh rw g ?)
r0 r* r r r rs r r< r r )r= r* r r cr s r
test_basiczTestSqueeze.test_basict s %%%%%% DRQ " " DQAr " " DAr2 2:a=="*Q *E*EFFF2:a=="*Q *E*EFFF2:a=="*Q*A*ABBB eWIjmmS# SXq!!!T#YY
+++++r" Nr` ra rb r rc r" r r r s s# , , , , ,r" r c e Zd Zd Zd Zej dej ej
g d Z d Z ej dg d d Z
d S )
TestKronc t j d }t j ddgddgg }t j ddgddgg }t t j || | t j ddgddgg }t j d }t t j || | t j dg }t j ddgddgg }t j ddgddgg }t t j || | t j ddgddgg }t j dg }t t j || | t j dggdggg }t j ddgddgg }t j ddgddggddgddggg }t t j || | t j ddgddgg }t j dggdggg }t j ddgddggddgddggg }t t j || | d S ) NrG r. r+ r, r r r, r )r rK r r )r= r r ks r r zTestKron.test_basic sR HQKK Hq!fq!f% & & Hq!fq!f% & &271a==!,,, Hq!fq!f% & & HQKK271a==!,,,
HaSMM Hq!fq!f% & & Hq!fq"g& ' '271a==!,,, Hq!fq!f% & & HaSMM271a==!,,,
HseqcU^ $ $ Hq!fq!f% & & H1v1v&!Q!Q(89 : :271a==!,,, Hq!fq!f% & & HseqcU^ $ $ H1v1v&!Q!Q(89 : :271a==!,,,,,r" c G d dt j }t j ddg } ||j |j |j }t
t t || t j t
t t || | t
t t || | t
t t || | d S )Nc e Zd ZdZdS )*TestKron.test_return_type..myarrayg ?N)r` ra rb __array_priority__rc r" r myarrayr s !$ r" r r. ) r r rH rV rF r r r r )r= r r r s r test_return_typezTestKron.test_return_type s % % % % %bj % % %
GQFOO
WQWagqv
.
.T$q!**%%rz222T$r2,,''111T$q"++&&000T$r1++&&00000r" array_classc |t j ddg } |t j ddg } |t j ddg }t t j || | d S )Nr+ r )r rH r r )r= r r r r s r test_kron_smokezTestKron.test_kron_smoke su
KA ( ( KA ( ( KA ( (271a==!,,,,,r" c h t j ddgddggddgddgg }t j t j g d t j t j d t
}t
|t j || d S ) NrG r. r+ r, r r )rG r, r, rE )r r rK diagidentityrN r r )r= r r s r test_kron_mazTestKron.test_kron_ma s EKK!Q!Q(AA/?K @ @ EKK
..hr{1~~T::::
<
< 1bgamm,,,,,r" zshape_a,shape_b))r r )r@ )r, r- r )r. r. )r. r. r. ))rG r r )r. r r. r ))r. r r r. r c t j | }t j | }dt dt | t | z
z |z }dt dt | t | z
z |z }t j || }t j || }t j |j | s
J d d S )N)rG r zUnexpected shape from kron)r rH r6 rz multiplyr array_equalrV ) r= shape_ashape_br r normalised_shape_anormalised_shape_bexpected_shaper s r test_kron_shapezTestKron.test_kron_shape s
GG GG !C3w<<G +D$E$E E O !C3w<<G +D$E$E E O%79KLL GAqMM~ ) ) G G*F G G ) G Gr" N)r` ra rb r r r. r/ parametrizer asarraymatr r r rc r" r r r s - - -: 1 1 1 [
BF+ - - -- - - [
G G G G Gr" r c & e Zd Zd Zd Zd Zd ZdS )TestTilec t j g d }ddgddgg}t t |d g d t t |d g dg dg t t |d g dg t t |d g d g d
g t t |d ddgddgddgddgg t t |d g d g d
g d g d
g d S ) Nr rG r. r+ r, )r rG r. r rG r. r rG r. )rG r. rG r. )r+ r, r+ r, r. rG )r rK r r
r= r r s r r zTestTile.test_basic s3 HYYY VaV T!QZZ!3!3!3444T!V__'9'9'9;M;M;M&NOOOT!V__'9'9'9&:;;;T!QZZ,,, !=>>>T!V__1v1v1v1v&FGGGT!V__|||\\\'3||\\\'C D D D D Dr" c t j d }t |d }|dz
}t |t j d d S )Nr- rG r. )r rr r
r r s r (test_tile_one_repetition_on_array_gh4679z1TestTile.test_tile_one_repetition_on_array_gh4679 sC IaLL AJJ QQ ! %%%%%r" c t j g gg }t j g g g }t |d j }t |d j }t |d t |d d S )Nr. r^ )r. r )r+ r. r )r rK r
rV r )r= r r r rx s r rY zTestTile.test_empty sp HrdV Hb"X AJJ I $Q Q """""r" c ddl m} g d}g d}|D ]_} |dd| }|D ]L}t j ||j }t || }t
|| } t || M`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 rD )size)r0 r r rH rF r
r r )
r= r repsrV r r rr largeklarges
r test_kroncomparezTestTile.test_kroncompare s ((((((@@@JJJ , ,A2A&&&A
,
,GAqw'' Q
a UF++++
, , ,r" N)r` ra rb r r rY r rc r" r r r sS D D D& & & # # # , , , , ,r" r c e Zd Zd ZdS )TestMayShareMemoryc
t j d }t j d }t t j || t t j ||d d d t t j ||d d d t t j ||dd d d df t t j |d d d | t t j |d d d | t t j |dd d d df | t t j |dd d d df | d S )N)rk ri )rg ri r r r. rG )r rH r may_share_memory)r= rx d2s r r z TestMayShareMemory.test_basic se GH
W[
!
!#Aq))***#Aq2w//000#Aq1v..///#AqTTrT{33444B'$$B$444555B'##A#333444B'!""ddd( R888999#Bqrr44R4xL"5566666r" Nr rc r" r r r s# 7 7 7 7 7r" r c t | t | k rt d t | | D ]\ }}t || dS )zCompare lists of arrays.z Iterables have different lengthsN)rz rJ zipr )r r r ys r r r
sb
3xx3w<< ;<<<C!! ! !11a ! !r" ).numpyr r# sysr. numpy.lib.shape_baser r r r r r r r
r r r
r r r
numpy.testingr r r r r maxsizer1 r% r' re ru r r r
r3 r9 rT rd rq rz r r r r r rc r" r r s
; 1. 1. 1. 1. 1. 1. 1. 1.h9 9 9 9 9 9 9 92c= c= c= c= c= c= c= c=LC C C C C C C C%5 %5 %5 %5 %5 %5 %5 %5PC& C& C& C& C& C& C& C&L
/
/
/
/
/
/
/
/ ; ; ; ; ; ; ; ;:"5 "5 "5 "5 "5 "5 "5 "5N& & & & & & & &:& & & & & & & &6 & & & & & & & &B, , , , , , , ,&LG LG LG LG LG LG LG LG^%, %, %, %, %, %, %, %,P 7 7 7 7 7 7 7 7 ! ! ! ! !r"