+
    eiP  c                   s$   ^ RI t ^ RIt^ RIHtHtHtHtHtH	t	H
t
HtHtHtHtHt ^ RIHtHtHtHtHtHtHt  ! R R4      t ! R R4      t ! R R	4      t ! R
 R4      t ! R R4      t ! R R4      tR t ! R R4      tR t  ! R R4      t!R t"R# )    N)c_diag_indicesdiag_indices_fromfill_diagonal	index_expix_mgridndenumeratendindexogridr_s_)assert_assert_almost_equalassert_array_almost_equalassert_array_equalassert_equalassert_raisesassert_raises_regexc                   s   a  ] tR t^t o R tR tR tR tR tR t	R t
]P                  P                  R. RO4      R	 4       tR
 tR tRtV tR# )TestRavelUnravelIndexc                s	   \        \        P                  ! ^R4      R4       \        \        P                  ! ^RR7      R4       \        \        4      ;_uu_ 4        \        P                  ! ^RR7       RRR4       \        \        4      ;_uu_ 4        \        P                  ! ^RR7       RRR4       \        \        4      ;_uu_ 4        \        P                  ! ^RR7       RRR4       \        \        4      ;_uu_ 4        \        P                  ! ^RR7       RRR4       \        \        P
                  ! RR4      ^4       \        \        P                  ! ^R4      R4       \        \        P
                  ! RR4      ^4       \        \        \        P                  RR4       \        \        \        P                  RR4       \        \        \        P                  ^R4       \        \        \        P
                  RR4       \        \        \        P
                  RR4       \        \        \        P
                  RR4       \        \        \        P
                  RR4       \        \        \        P
                  RR4       \        \        P                  ! ^.R4      . RO4       \        \        P
                  ! . ROR4      ^.4       \        P                  ! . RO. RO.4      p\        \        P
                  ! VR4      . RO4       \        \        P
                  ! VRRR	7      . RO4       \        \        P
                  ! VRR
R7      . RO4       \        \        P
                  ! VR R!R7      . R"O4       \        \        P
                  ! R#R$4      R4       \        \        P                  ! \        P                  ! . RO4      R4      . RO. RO.4       \        \        P                  ! \        P                  ! . RO4      RRR	7      . RO. RO.4       \        \        P                  ! RR$4      . R#O4       R#   + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     ELf; i)%   )indicesshape)r   hapeN)r   )Zims)Zdimsg      ?FZorderclipmodeiU  )r   r      r    )i   i^   )r   iB   )r   )r   r   )r    r!   )r    r   )皙?g        )         )r   r   r#   )r$   r%   r%   )r#      r   )   r%   )   )   i%   )i   r)      )r#   r%   )r(   i   i   r#   r#   )r   wrap)   r*   r*   )r$   r   r#   r   )r%   r'      	   )r   npunravel_indexr   	TypeErrorravel_multi_index
ValueErrorarray)selfarr   & H/usr/lib64/python3.14/site-packages/numpy/lib/tests/test_index_tricks.py
test_basicZ TestRavelUnravelIndex.test_basic   s   R%%a0&9 	R%%a,24&,	. 9%%QV4 & 9%%QV, & 9%%Sh/ & 9%%Sx0 & 	R))&&91=R%%c84g>R))'8<cBj""2"2B?i!1!13?j""2"2Av>j""6"6Hj""6"6Gj""6"6Hj""6"6Gi!5!5y&IR%%&99EyQ  I68K	M hh	9-.R))#v6E  fC8,	H  f6:L	JR))#v<LM!	#R)),EtLR%%bhh|&<fE+	-RXXl3V3G	"	$ 	R%%dL9<HW &%% &%% &%% &%%s0   P=QQ%Q9=Q	Q"	%Q6	9R
	c           
     s   R pRp\        \        V\        P                  . R4       \        \        V\        P                  RR4       \        \        V\        P                  \        P                  ! . 4      R4       \        \        P                  ! \        P                  ! . \        R7      R4      . . . .4       \        \        V\        P                  . . 3R4       \        \        V\        P                  . R.3R4       \        \        V\        P                  \        P                  ! . 4      \        P                  ! . 4      3R4       \        \        P                  ! \        P                  ! . \        R7      \        P                  ! . \        R7      3R4      . 4       \        \        P                  ! \        P                  ! . . .\        R7      R4      . 4       R# )	z9indices must be integral: the provided empty sequence waszonly int indices permitteddtypeZabcN)
   r$   r&    r=   r$   )r&   r$   )r   r2   r0   r1   r5   r   intr3   )r6   Zmsg1Zmsg2   &  r9   test_empty_indicesZ(TestRavelUnravelIndex.test_empty_indicesU   sT   J+ItR-=-=r:NItR-=-=r:NItR-=-=rxx|&	(R%%bhhr&=zJ"b\	#ItR-A-AB8#	%ItR-A-AB=#	%ItR-A-AXXb\288B<0&	:R))"C("((2S*ABFLMO	QR))"((B83*G 	"    c                s   \         P                  \         P                  8X  d;   ^^.^^.^^u.^^.RR.^^.^ ^.3p\        \         P                  ! VR4      RR.4       \        \        \         P                  ^R	4       ^ .^ .3p\         P                  ! \         P                  4      P                  ^,          p\        \         P                  ! W#^34      ^ .4       \        \        \         P                  W#^,           ^34       \        \         P                  ! W#^3RR7      ^ .4       \        \        \         P                  W#^,           ^3RR7       R# )
r   iK	  i  l   z l   nm r   r   N)r)   r'   ix   $   i
  r.   r%   )l    l       )
r0   intpint64r   r3   r   r4   r1   ZiinfoZmax)r6   r7   Z	dummy_arrZhalf_max   &   r9   test_big_indicesZ&TestRavelUnravelIndex.test_big_indicesi   s)   77bhhr7QFQHr1g$<!Q!Q1C$$S*FG\*,
 	j""2"2A7MN S1#J	88BGG$((A-  qM:QC	Aj  )lA->	@  qMEs	Lj  )lA->c	KrC   c           
     s   \         P                  \         P                  \         P                  \         P                  \         P
                  \         P                  3 EF  p\         P                  ! . RO. RO.VR7      pRp^V^ ,          ,          V^,          ,           p\        \         P                  ! W#4      V4       \        V\         P                  ! WC4      4       V^ ,          ^V^,          ,          ,           p\        \         P                  ! W#RR7      V4       \        V\         P                  ! WCRR7      4       \         P                  ! . RO. RO. RO.VR7      pR	p^
^V^ ,          ,          V^,          ,           ,          V^,          ,           p\        \         P                  ! W#4      V4       \        V\         P                  ! WC4      4       V^ ,          ^V^,          ^V^,          ,          ,           ,          ,           p\        \         P                  ! W#RR7      V4       \        V\         P                  ! WCRR7      4       EK  	  R# )
r   r;   r   r   N)r   r    r   r   r$   r#   )r   r%   r   r$   r   r    )r&   r.   )r   r$   r   r    r/   r&   )r&   r.   r=   )r0   Zint16Zuint16Zint32uint32rF   Zuint64r5   r   r3   r1   )r6   r<   Zcoordsr   Zuncoords   &    r9   test_dtypesZ!TestRavelUnravelIndex.test_dtypes   s   hh		288ii2996EXX#%78GFE6!9}vay0H--f<hG!1!1(!BCay1vay=0H$$V#>J!1!1(!MNXX#%79KLF EQ]VAY67&)CH--f<hG!1!1(!BCay1q	Aq	M(A#BBH$$V#>J!1!1(!MN/6rC   c                s:   \        \        P                  ! . RORRR7      \        P                  ! . ROR4      4       \        \        P                  ! . RORRR7      \        P                  ! . ROR4      4       \        \        \        P                  . ROR4       R# )	r&   r,   r   N)r&   r   r    r   )r#   r$   r'   r-   )r   r   r%   r   )r,   raiser   rL   )r   r   r    r   )r   r0   r3   r   r4   r6      &r9   test_clipmodesZ$TestRavelUnravelIndex.test_clipmodes   s{      FK  }=	? 	R))-0PQ )),F	H 	,,m]	LrC   c                s    \         P                  ! . ROR4      w  r\        VP                  P                  4       \        VP                  P                  4       R# )r   Nr   r   r$   )r#   r&   )r0   r1   r   ZflagsZ	writeable)r6   xyrA   r9   test_writeabilityZ'TestRavelUnravelIndex.test_writeability   s:    	62!!"!!"rC   c                s    \         P                  ! ^ R4      p\        VR4       \        \        R\         P                  ^ .R4       \        \        R\         P                  ^.R4       R# )r    z0d arrayzout of boundsNr>   )r0   r1   r   r   r4   r6   rQ   r8   r9   test_0dZTestRavelUnravelIndex.test_0d   sR    Q#QJ
B4D4Dqc2N)9)9A3	DrC   r   c           	     s   \         P                  ! \         P                  ! R\         P                  R7      RVR7      pVP                  R8X  g   Q h\        \        4      ;_uu_ 4        \         P                  ! \         P                  ! R\         P                  R7      RVR7       RRR4       R#   + '       g   i     R# ; i)r$   r;   r   N)r$   r    r   r   r    r    )r$   r   )r0   r3   zerosrE   r   r   r4   )r6   r   ress   && r9   test_empty_array_ravelZ,TestRavelUnravelIndex.test_empty_array_ravel   sz    ""HHV2773YTKyyD   :&&  HHV2773YTK '&&&s   (>B00C	c                s   \         P                  ! \         P                  ! ^ \         P                  R7      R4      p\	        V4      ^8X  g   Q h\
        ;QJ d    R V 4       F  '       d   K   RM	  RM! R V 4       4      '       g   Q h\        \        4      ;_uu_ 4        \         P                  ! ^.R4       RRR4       R#   + '       g   i     R# ; i)r    r;   c              3   s>   "   T F  qP                   R8H  x  K  	  R# 5i)r    NrW   )r   ).0ar8   r9   	<genexpr>ZATestRavelUnravelIndex.test_empty_array_unravel.<locals>.<genexpr>   s     0Cq77d?Cs   FTNrV   )r0   r1   rX   rE   Zlenallr   r4   )r6   rY   r8   r9   test_empty_array_unravelZ.TestRavelUnravelIndex.test_empty_array_unravel   s    rxx99E3x1}}s0C0sss0C00000:&&aS), '&&&s   !CC	c                s    \         P                  ! \         P                  ! . RO.\         P                  R7      R4       R# )r   r;   N)r   r    r   r    )r#   )r0   r1   r5   rI   rM   rN   r9   test_regression_size_1_indexZ2TestRavelUnravelIndex.test_regression_size_1_index   s%     	<.		BDIrC   r>   N)r   r,   rL   )__name__
__module____qualname____firstlineno__r:   rB   rH   rK   rO   rS   rU   pytestmarkparametrizerZ   r_   r`   __static_attributes____classdictcell__Z__classdict__   @r9   r   r      sh     5In"(K0O6
L#D [[V%>?K @K-J JrC   r   c                   s   a  ] tR t^t o R tR tR tR t]P                  P                  RRR.4      R 4       tR tR	 tR
 tRtV tR# )TestGridc                s   \         R^R1,          p\         R^R1,          p\        VP                  R8H  4       \        VP                  R8H  4       \        V^ ,          R8H  4       \        VR,          ^4       \        V^ ,          R8H  4       \        V^,          V^ ,          ,
          R^4       \        VR,          V^ ,          R,           ^4       \        V^,          V^ ,          ,
          R^4       R# )	r                 $@r"   Nr    )r=   )   ggfffff?gqq?)r   r   r   r   )r6   r\   brA   r9   r:   ZTestGrid.test_basic   s    "Qs(O"Qs(O5 !5 !!
AbE1%!
AaD1Q4Kb1AbE1Q4(?B7AaD1Q4KB7rC   c                s    \         P                  ! ^^
RR7      w  r\        VR4       \        V\        R,          ^4       R# )r   T)Zretstep:r   r=   y              I@Ng9/?)r0   Zlinspacer   r   r   )r6   rR   ZstrA   r9   test_linspace_equivalenceZ"TestGrid.test_linspace_equivalence   s0    Ar40B)!!U8_b9rC   c                sL   \         R^R1R^R13,          p\         R^R1R^R13,          p\        VP                  R8H  4       \        VP                  R	8H  4       \        V^ ,          R
,          \        P
                  ! ^
R4      ) 4       \        V^,          R,          R\        P
                  ! ^
R4      ,          4       \        V^ ,          R,          \        P
                  ! ^
R4      ^4       \        V^,          R,          ^\        P
                  ! ^
R4      ,          ^4       \        VR,          VR,          ,
          R\        P
                  ! ^R4      ,          ^4       \        VR,          VR,          ,
          R\        P
                  ! ^R4      ,          ^4       R# )r   rm   r"   g?dNr    )r   r=   r=   )r   rn   rn   )r    NNN)rs   r    )r    rs   )rs   r    )r    r   rs   )r    r    rs   )r   rs   r   )r   rs   r    )r   r   r   r   r0   onesr   )r6   crq   rA   r9   test_ndZTestGrid.test_nd   s7   "Qs(BqH$%"Qs(BqH$%;&';&'1Q4:C(8'891Q4:rBGGB,<'<=!!A$u+rwwr3/?D!!A$u+q2772s3C/CRH!!G*qz"9"%C(8"8"	>!!G*qz"9"%C(8"8"	>rC   c                s    \         R^R1R^R13,          p\        R^R1R^R13,          p\        P                  ! V!  p\	        W4       F  w  rE\        WE4       K  	  R# )r   rm   Nr    rr   )r   r
   r0   Zbroadcast_arrayszipr   )r6   Z	grid_fullZgrid_sparseZgrid_broadcastZfro   s   &     r9   test_sparseZTestGrid.test_sparse   sc    "Qs(BqH,-	BqHb3h./ ,,k:	2DA 3rC   zstart, stop, step, expectedNc                s    \         WV1WV13,          p\         WV1,          p\        VP                  V^ ,          4       \        VP                  V^,          4       R# )r    N)r   r   Zsize)r6   ZstartZstopZstepexpectedZgridZ
grid_smalls   &&&&&  r9   test_mgrid_size_none_handlingZ&TestGrid.test_mgrid_size_none_handling   sO     U_eo56 5d?+
TYY,Z__hqk2rC   c                sh   \         R,          p\         \        P                  ! R4      \        P                  ! R4      \        P                  ! R4      13,          p\        W4       VP                  \        P                  8X  g   Q hVP                  \        P
                  8X  g   Q h\         R ,          p\         \        P                  ! R4      \        P                  ! R4      \        P                  ! R4      1,          p\        VP                  \        P
                  8H  4       \        W4       R# )r"   Q?r"   r"   r|   Nr{   )r   r0   float32r   r<   float64r   )r6   grid64Zgrid32rA   r9   test_accepts_npfloatingZ TestGrid.test_accepts_npfloating  s    }&rzz#rzz$'7

3GHJ!&1||rzz)))||rzz))) |$rzz#rzz$'7

3GH

*+!&1rC   c                sh   \         R,          p\         \        P                  ! R4      \        P                  ! R4      \        P                  ! R4      13,          p\        VP                  \        P                  8H  4       \        W4       \         ^ \        P                  ! ^4      R1,          p\         ^ \        P                  ! ^4      R13,          p\        VP                  VP                  u;8H  ;'       d    \        P                  8H  Mu 4       \        W4^ ,          4       \         R ,          p\         \        P                  ! R4      \        P                  ! R4      \        P                  ! R4      1,          p\        VP                  \        P                  8H  4       \        W4       R# )r{   r"   r|   y        333333@Nr}   )r   r0   
longdoubler   r<   r   r   )r6   r   Zgrid128Z
grid128c_aZ
grid128c_brJ   r9   test_accepts_longdoubleZ TestGrid.test_accepts_longdouble  s   }&MM#r}}T22==3EEF
 	./!&21R]]1-d23
1R]]1-d235

  J$4$4EEEF:!}5 |$MM#r}}T22==3EE
 	./!&2rC   c           	     s   \        \        R,          \        RR\        P                  ! R4      13,          4       \        \        R ,          \        RR\        P                  ! R4      1,          4       \        R,          p\        R,          ^ ,          p\	        VP
                  VP
                  u;8H  ;'       d    \        P                  8H  Mu 4       \        W4       \        RR\        P                  ! R4      1,          p\        RR\        P                  ! R4      13,          ^ ,          p\	        VP
                  VP
                  u;8H  ;'       d    \        P                  8H  Mu 4       \        W4       R# )	r"   333333?              @r"   r   r   r"   r           ffffff
@r   N)r   )r   )
r   r   r0   	complex64r   r<   r   r   Zclongdoubler   )r6   Zgrid64_aZgrid64_bZ	grid128_aZ	grid128_brJ   r9   test_accepts_npcomplexfloatingZ'TestGrid.test_accepts_npcomplexfloating-  s    !+s3r||B/?'?'@!B	

 	"*uSR\\"-=%=>	

 &(+(..>>BJJ>?8.#c".."667	#c".."6679!<		9??CCbmmCD8.rC   r>   )Nr=   rm   )i   r=   )irn   N)i  i   )ra   rb   rc   rd   r:   rp   rv   rx   re   rf   rg   rz   r   r   r   rh   ri   rj   rk   r9   rl   rl      s^     
8:
> [[:"#=
 
3	
323,/ /rC   rl   c                   sD   a  ] tR tRt o R tR tR tR tR tR t	Rt
V tR	# )
TestConcatenatoriD  c                s    \        \        R,          \        P                  ! . RO4      4       \        P                  ! ^4      p\        V^ ^ V3,          p\        V. RO4       R# )r   N)r   r   r$   r#   r&   r%   )r   r   r   r   r   r    r    r   r   r   r   r   )r   r   r0   r5   rt   )r6   ro   ru   rA   r9   test_1dZTestConcatenator.test_1dE  sE    2./:L1MNGGAJq!QzN1BCrC   c                sP    \         R,          p\        VP                  R8H  4       R# )333333$@f8N)r   :r   r=   N)r   r   r<   r6   Zgr8   r9   test_mixed_typeZ TestConcatenator.test_mixed_typeK  s    zN4 rC   c                s    \         R\        P                  ! ^.4      \        P                  ! . RO4      R3,          p\        VP                  R8H  4       R# )r   g      $@r   Ng333333$)r   r$   r#   )r   r0   r5   r   r<   r   r8   r9   test_more_mixed_typeZ%TestConcatenator.test_more_mixed_typeO  s8    ubhhsmRXXi%8$>?4 rC   c                s    \         R ,          p\        VP                  R8H  4       \         ^ ^$\        P                  ! R4      1,          p\        VP                  R8H  4       R# ):r    rD                 Y@r   N)d   )r   r   r   r0   r   r   r8   r9   test_complex_stepZ"TestConcatenator.test_complex_stepS  sI    yM6!" qBLL&&'6!"rC   c                s   \         P                  P                  ^^4      p\         P                  P                  ^^4      p\        RW3,          p\	        VP
                  R8H  4       \        VR,          V4       \        VR,          V4       \        W3,          p\	        VP
                  R8H  4       \        VR,          V4       \        VR,          V4       R# )	r&   Z1N)r&   r=   )rs   Nr&   N)rs   r&   NN)r=   r&   )r   rs   )r   rs   )r0   randomrandr   r   r   r   )r6   ro   ru   rq   rG   r9   test_2dZTestConcatenator.test_2d\  s    IINN1a IINN1a sAyM7"#1U8Q'1U8Q'qtH7"#1U8Q'1U8Q'rC   c                s*   \        \        ^ \        P                  ! ^4      ^3,          . RO4       \        \        . RO\        P                  ! ^4      3,          . RO4       \        \        \        P                  ! ^ 4      . RO3,          . RO4       R# )r    Nr    r   r   )r    r   r   r$   rP   )r   r   r0   r5   rM   rN   r9   rU   ZTestConcatenator.test_0dh  sX    R288A;)*I6R	288A;./>RY./>rC   r>   N)ra   rb   rc   rd   r   r   r   r   r   rU   rh   ri   rj   rk   r9   r   r   D  s)     D!!#
(? ?rC   r   c                   s&   a  ] tR tRt o R tRtV tR# )TestNdenumeratein  c                s~    \         P                  ! ^^.^^..4      p\        \        \	        V4      4      . RO4       R# )r   N)))r    r    r   ))r    r   r   )r   r$   ))r   r   r#   )r0   r5   r   listr   r6   r\   r8   r9   r:   ZTestNdenumerate.test_basico  s4    HHq!fq!f%&T+a.)I	KrC   r>   N)ra   rb   rc   rd   r:   rh   ri   rj   rk   r9   r   r   n  s     K KrC   r   c                   s,   a  ] tR tRt o R tR tRtV tR# )TestIndexExpressioniu  c                s    \         P                  ! ^4      p\        VRR V\        RR ,          4       \        VRR V\        RR ,          4       R# )r   Nr    )r0   aranger   r   r   r   r8   r9   test_regression_1Z%TestIndexExpression.test_regression_1v  sD    IIaLQsVQr#2wZ(QsVQy"~./rC   c           	     s   \         P                  P                  ^^^4      p\        VRR^^.3,          V\        RR^^.3,          ,          4       \        VRR^^.3,          V\
        RR^^.3,          ,          4       R# )r#   rs   :Nr$   NN)r0   r   r   r   r   r   r   r8   r9   test_simple_1Z!TestIndexExpression.test_simple_1|  sr    IINN1a#Qq"q!f}%q1b1a&=)A'BCQq"q!f}%qArAq6M):';<rC   r>   N)ra   rb   rc   rd   r   r   rh   ri   rj   rk   r9   r   r   u  s     0= =rC   r   c                   s>   a  ] tR tRt o R tR tR tR tR tRt	V t
R# )	TestIx_i  c                s   \         P                  ! \        ^ 4      4      w  p\        VP                  \         P
                  4       \         P                  ! . 4      w  p\        VP                  \         P
                  4       \         P                  ! \         P                  ! . \         P                  R7      4      w  p\        VP                  \         P                  4       R# )r    r;   N)r0   r   ranger   r<   rE   r5   r~   r   r8   r9   r   ZTestIx_.test_regression_1  s|    VVE!HQWWbgg&VVBZQWWbgg& VVBHHRrzz23QWWbjj)rC   c           
     sb  a Rp\         \        P                  3 EF  p\        P                  ! V Uu. uF
  q2! V4      NK  	  up!  p\	        \        WA4      4       F  w  ow  rS\        VP                  S,          V4       \        \        ;QJ d2    V3R l\	        VP                  4       4       F  '       d   K   RM'	  RM#! V3R l\	        VP                  4       4       4      4       \        \        P                  ! VP                  \        P                  4      4       K  	  EK  	  R# u upi )r#   c              3   sD   <"   T F  w  rVS8w  g   K  V^8H  x  K  	  R# 5i)r   Nr>   )r[   ZjZshks   &  r9   r]   Z/TestIx_.test_shape_and_dtype.<locals>.<genexpr>  s!     N1CqAvGB!G1Cs     FTN)r#   r&   r$   r   )r   r0   r   r   Z	enumeraterw   r   r   r   r^   Z
issubdtyper<   Zinteger)r6   ZsizesZfuncZszZarraysr\   r   s   &     @r9   test_shape_and_dtypeZTestIx_.test_shape_and_dtype  s    BII&DVV72d2h78F'F(:;
7AQWWQZ,N1771CNN1771CNNOaggrzz:; < '7s   D,
c                s    . ROp\         P                  ! V4      w  p\        \         P                  ! V4      ^ ,          V4       R# )TN)TFTT)r0   Znonzeror   r   )r6   Zbool_aZint_arA   r9   	test_boolZTestIx_.test_bool  s.    *F#RVVF^A&.rC   c                sT    . RO. RO.p\        \        \        P                  V4       R# )r   NrP   r#   r&   r%   )r   r4   r0   r   )r6   Zidx2dr8   r9   test_1d_onlyZTestIx_.test_1d_only  s    I&j"&&%0rC   c                s    ^p\         P                  ! V4      p\        W"4      p\        V^ ,          P                  V^34       \        V^,          P                  ^V34       \        VP                  V34       R# )r&   N)r0   r   r   r   r   )r6   Zlength_of_vectorrQ   ZoutrG   r9   test_repeated_inputZTestIx_.test_repeated_input  sb    II&'!iSV\\$4a#89SV\\A'7#89QWW/12rC   r>   N)ra   rb   rc   rd   r   r   r   r   r   rh   ri   rj   rk   r9   r   r     s#     
*</
13 3rC   r   c                  s    \         \        P                  ! . RO.4      ^ ^ \        P                  ! . RO.4      3,          p \        V . RO.4       R# )r   NrP   r   )r   r   r$   r    r    r#   r&   r%   )r   r0   r5   r   )r\   s    r9   test_c_r     s9    
288YK !Q)(==>A-./rC   c                   sJ   a  ] tR tRt o R tR tR tR tR tR t	R t
R	tV tR
# )TestFillDiagonali  c           	         \         P                  ! R\        4      p\        V^4       \	        V\         P
                  ! . RO. RO. RO.4      4       R# )r$   N)r$   r$   r&   r    r    r    r&   r    r    r    r&   r0   rX   r@   r   r   r5   r   r8   r9   r:   ZTestFillDiagonal.test_basic  s?    HHVS!arxx""$ %	rC   c                s    \         P                  ! R\        4      p\        V^4       \	        V\         P
                  ! . RO. RO. RO. RO. RO. RO. RO. RO. RO. RO.
4      4       R# )r=   Nr?   r   r   r   r    r    r    r   r   r8   r9   test_tall_matrixZ!TestFillDiagonal.test_tall_matrix  sT    HHWc"arxx"""""""""	$ 	%	rC   c                s    \         P                  ! R\        4      p\        V^R4       \	        V\         P
                  ! . RO. RO. RO. RO. RO. RO. RO. RO. RO. RO.
4      4       R# )r=   TNr?   r   r   r   r   r   r   r8   r9   test_tall_matrix_wrapZ&TestFillDiagonal.test_tall_matrix_wrap  sV    HHWc"aD!rxx"""""""""	$ 	%	rC   c           	     r   )r$   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   r8   r9   test_wide_matrixZ!TestFillDiagonal.test_wide_matrix  s@    HHWc"arxx7779 :	rC   c                s    \         P                  ! R\        4      p\        V^4       \         P                  ! . RO4      p\        \         P                  ! V^ 8g  4      W"W"34       R# )r$   N)r$   r$   r$   r$   r   )r0   rX   r@   r   r5   r   Zwhere)r6   r\   ZirA   r9   test_operate_4d_arrayZ&TestFillDiagonal.test_operate_4d_array  sE    HH\3'aHHYRXXa1f%a|4rC   c                s    \         P                  ! ^\        4      p\        \        R4      ;_uu_ 4        \        V^4       RRR4       R#   + '       g   i     R# ; i)r$   at least 2-dNr0   rX   r@   r   r4   r   r   r8   r9   test_low_dim_handlingZ&TestFillDiagonal.test_low_dim_handling  s8    HHQ ^<<!Q =<<<   AA	c                s    \         P                  ! R\        4      p\        \        R4      ;_uu_ 4        \        V^4       RRR4       R#   + '       g   i     R# ; i)r$   equal lengthN)r$   r$   r'   r$   r   r   r8   r9   test_hetero_shape_handlingZ+TestFillDiagonal.test_hetero_shape_handling  s:     HH\3' ^<<!Q =<<<r   r>   N)ra   rb   rc   rd   r:   r   r   r   r   r   r   rh   ri   rj   rk   r9   r   r     s-       5    rC   r   c            
      sr   \        ^4      p \        P                  ! . RO. RO. RO. RO.4      p^dW&   \        V\        P                  ! . RO. RO. RO. R	O.4      4       \        ^^4      p\        P                  ! R
\
        4      p^W&   \        V\        P                  ! ^^ .^ ^ ..^ ^ .^ ^...4      4       R# )r#   N)r   r   r$   r#   )r&   r%   r'   r.   )r/   r=   i   r-   )r*         i   )r   r   r$   r#   )r&   r   r'   r.   )r/   r=   r   r-   )r*   r   r   r   )r   r   r   )r   r0   r5   r   rX   r@   )Zdir\   Zd3s      r9   test_diag_indicesr     s    	aB
,!"$ 	%A AE	288^#%&( )
 
a	B 	C AAE	288q!f!f!f!f  
rC   c                   s2   a  ] tR tRt o R tR tR tRtV tR# )TestDiagIndicesFromi  c                s    \         P                  P                  R4      p\        V4      w  r#\        V\         P                  ! ^4      4       \        V\         P                  ! ^4      4       R# )r#   Nr+   )r0   r   r   r   r   )r6   rQ   Zrru   rG   r9   test_diag_indices_fromZ*TestDiagIndicesFrom.test_diag_indices_from  sF    IIV$ #1biil+1biil+rC   c                s    \         P                  ! ^4      p\        \        R4      ;_uu_ 4        \	        V4       RRR4       R#   + '       g   i     R# ; i)r'   r   N)r0   rt   r   r4   r   rT   r8   r9   test_error_small_inputZ*TestDiagIndicesFrom.test_error_small_input  s3    GGAJ ^<<a  =<<<s   AA	c                s    \         P                  ! R\        4      p\        \        R4      ;_uu_ 4        \        V4       RRR4       R#   + '       g   i     R# ; i)r$   r   N)r$   r$   r   r$   )r0   rX   r@   r   r4   r   rT   r8   r9   test_error_shape_mismatchZ-TestDiagIndicesFrom.test_error_shape_mismatch"  s6    HH\3' ^<<a  =<<<s   A

A	r>   N)	ra   rb   rc   rd   r   r   r   rh   ri   rj   rk   r9   r   r     s     ,!
! !rC   r   c                  s   \        \        ^^^4      4      p \        \        P                  ! R4      4       UUu. uF  w  rVNK	  	  ppp\        W4       \        \        R4      4      p \        W4       \        \        R4      4      p \        V \        \        ^4      4      4       \        \        4       4      p \        V R.4       \        \        R4      4      p \        V R.4       \        \        ^ .!  4      p \        V . 4       R# u uppi )r   NrP   )r$   r>   )r   r	   r   r0   rX   r   r   )rQ   ZixZery   s       r9   test_ndindexr   (  s    WQ1A +BHHY,? @A @ur @HAq#WY Aq# 	WT]Aq$wqz*+ 	WYARDWR[ARD 	Wqc]AB' Bs   D)#re   Znumpyr0   Znumpy.lib._index_tricks_implr   r   r   r   r   r   r   r   r	   r
   r   r   Znumpy.testingr   r   r   r   r   r   r   r   rl   r   r   r   r   r   r   r   r   r   r>   rC   r9   <module>r      s          pJ pJdr/ r/j'? '?TK K= ='3 '3T0
D  D N
8! !&rC   