+
    Pj(              	       b   ^ RI t ^ RIt^ RIt^ RIt^ RIHtHtHtH	t	 ^ RI
Ht ]P                  P                  4       tR tR R ltR R lt ! R	 R
4      tRR lt] P(                   ! R R4      4       t]! RR4      ]! RR4      ]! RR4      ]! RR4      ]! RR4      ]! RR4      .tR tR tR tR# )    N)EinopsErrorasnumpypackunpack)collect_test_backendsc                     \        W4      w  r#\        WV4      p\        V4      \        V 4      8X  g   Q h\        W@4       F6  w  rV\        P
                  ! \        V4      \        V4      4      '       d   K6  Q h	  R # N)r   r   lenzipnpallcloser   )xspatternxpsunpackedabs   &&     r/Users/ahmed/devFolder/Ultron/claude-voice/gateway/.venv/lib/python3.14/site-packages/einops/tests/test_packing.pypack_unpackr      s`    EAbg&Hx=CG###H!{{71:wqz2222 "    c                $    V ^8  d   QhR\         /#    r   str)formats   "r   __annotate__r      s      C r   c                     \        WV4      p\        W2R 7      w  rE\        P                  ! \	        V4      \	        V 4      4      '       g   Q hV# )r   )r   r   r   r   r   )r   r   r   r   packedps2s   &&&   r   unpack_and_packr#      s>    aW%Hx1KF;;wv
3333Or   c                $    V ^8  d   QhR\         /# r   r   )r   s   "r   r   r      s     . .# .r   c                 8   \        4       p\        4       pV;_uu_ 4        \        WV4      p\        WRR 7      w  rgRRR4       V;_uu_ 4        \        V 4      p\        WV4      p	\        WR 7      w  rRRR4       \	        VP
                  4      \	        VP
                  4      8X  g   Q hVP
                  e   R# \        P                  ! \        X4      \        V 4      4      '       g   Q h\        P                  ! \        X
4      \        V 4      4      '       g   Q h\        X4      \        X	4      8X  g   Q h\        WY4       F-  w  r\        P                  ! \        V4      V4      '       d   K-  Q h	  R#   + '       g   i     ELC; i  + '       g   i     EL; i)r    N)
CaptureExceptionr   r   r   type	exceptionr   r   r
   r   )r   r   r   capturer_backendcapturer_numpyr   r!   r"   x_npunpacked_np	packed_npps3r   r   s   &&&           r   unpack_and_pack_against_numpyr/      s3   ')%'N		!)85 
 
qzTw/k;	 

 (()T2B2L2L-MMMM+ {{76?GAJ7777{{79-wqz::::8}K 0000.DA;;wqz1---- /% 
		 
s   E4&F4F	F	c                   ,   a  ] tR t^7t o R tR tRtV tR# )r&   c                    R V n         R # r	   r(   selfs   &r   	__enter__CaptureException.__enter__8   s	    r   c                    W n         R # )Tr2   )r4   exc_typeexc_valexc_tbs   &&&&r   __exit__CaptureException.__exit__;   s
     r   r2   N)__name__
__module____qualname____firstlineno__r5   r;   __static_attributes____classdictcell____classdict__s   @r   r&   r&   7   s      r   r&   c           	         R  pR pV! ^W4      w  rEpV! W^ 4      pV! \         P                  ! WEV.^R7      \        WEV.R4      ^ ,          4       V! \         P                  ! WEV.^R7      \        WEV.R4      ^ ,          4       V! \         P                  ! WEV.^ R7      \        WEV.R4      ^ ,          4       V! \         P                  ! WEV.^R7      \        WEV.R4      ^ ,          4       V! \         P                  ! WEV.^ R7      \        WEV.R4      ^ ,          4       \         P                  R,          pV! \         P                  ! WH,          WX,          Wh,          V.^R7      \        WEWg.R4      ^ ,          4       \
        P                  ! \        4      ;_uu_ 4        \        WEWg.R	4       RRR4       \        WEV.R
4       \
        P                  ! \        4      ;_uu_ 4        \        WEWg.R4       RRR4       \
        P                  ! \        4      ;_uu_ 4        \        WEWg.R4       RRR4       \
        P                  ! \        4      ;_uu_ 4        \        WEWg.R4       RRR4       \
        P                  ! \        4      ;_uu_ 4        \        WEWg.R4       RRR4       \
        P                  ! \        4      ;_uu_ 4        \        WEWg.R4       RRR4       \        WEWg.R4       R#   + '       g   i     ELL; i  + '       g   i     EL; i  + '       g   i     L; i  + '       g   i     L; i  + '       g   i     L; i  + '       g   i     L; i)c                  ,    \         P                  V 4      # r	   )rngrandom)shapes   *r   rand test_numpy_trivial.<locals>.randA   s    zz%  r   c                     V P                   VP                   8X  g   Q hV P                  VP                  8X  g   Q h\        P                  ! W8H  4      '       g   Q hR # r	   )dtyperI   r   all)r   r   s   &&r   check!test_numpy_trivial.<locals>.checkD   sB    ww!''!!!ww!''!!!vvaf~~~r   )axiszh w *zh * wz* h wzh *z* wNzh w nonexisting_axis *z"some_name_for_H some_name_for_w1 *zh _w *zh_ w *z1h_ w *z1 w *zh h *zh H *)NNNrR   N)r   stackr   concatenate	index_exppytestraisesr   )	HWrJ   rO   rgr   
embeddingsis	   &&       r   test_numpy_trivialr^   @   s   !
 1amGA!aBJ	
!#aAY # 

!#aAY # 

!#aAY #
 

ayq)aAYq! 

ayq)aAYq!
 	Z A	
adAD*5A>aA"G,Q/
 
{	#	#aA"$<= 
$ 	!89	{	#	#aA"H- 
$	{	#	#aA"H- 
$	{	#	#aA"I. 
$	{	#	#aA"G, 
$	{	#	#aA"G, 
$ 	!	(! 
$	#	#
 
$	#	#	#	#	#	#	#	#	#	#sH   !K3(L!LL.MM3L	L	L+	.L>	M	M$	c                   8   a  ] tR t^{t o R tR tV 3R ltRtV tR# )UnpackTestCasec                T    V P                   P                  4       P                  R 4      # )*)r   splitindexr3   s   &r   dimUnpackTestCase.dim   s     ||!!#))#..r   c                V    V P                   V P                  4       ,          ^8X  g   Q hR# )   N)rI   re   r3   s   &r   	selfcheckUnpackTestCase.selfcheck   s    zz$((*%***r   c                Z   < V ^8  d   Qh/ S[ P                  S[R3,          ;R&   S[;R&   # )r   .rI   r   )typingTupleintr   )r   rD   s   "r   r   UnpackTestCase.__annotate__{   s)     <<S!!  L r    N)	r=   r>   r?   r@   re   ri   __annotate_func__rA   rB   rC   s   @r   r`   r`   {   s     
/+  r   r`   rb   z* sevenzseven *z* three fourzfour * threezthree four *c            
      
  a \          EF  p V P                  pV P                  p\        P	                  V4      o\        S^.^.^..V4       \        P                  ! \        4      ;_uu_ 4        \        S^.^.^..VR,           4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        S^.^.^..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        S^.^.^..V4       RRR4       \        S^.^.R..V4       \        S^.R.^..V4       \        SR.^.^..V4       \        S^.^.R..V4      w   r4VP                  V P                  4       ,          ^ 8X  g   Q h\        P                  ! \        4      ;_uu_ 4        \        S^.^.R..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        SR.^.^..V4       RRR4       \        S^^.^^.R^..V4      p\        ;QJ d    V3R lV 4       F  '       d   K   RM	  RM! V3R lV 4       4      '       g   Q h\        S^^.^R.^^..V4      p\        ;QJ d    V3R lV 4       F  '       d   K   RM	  RM! V3R lV 4       4      '       g   Q h\        S^R.^^.^^..V4      p\        ;QJ d    V3R lV 4       F  '       d   K   RM	  RM! V3R lV 4       4      '       g   Q h\        P                  ! \        4      ;_uu_ 4        \        SR^.^.^..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        S^^.^.^R..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        S^^.^.^R..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        S^^.^R.^..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        S^R.^^.^..V4       RRR4       \        S^ .^.R..V4       \        S^ .R.^..V4       \        SR.^.^ ..V4       \        S^R.^^..V4       EK  	  R#   + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL%; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     ELp; i  + '       g   i     ELD; i  + '       g   i     EL; i)	r   z non_existent_axisNc              3      <"   T F7  p\        VP                  4      \        SP                  4      ^,           8H  x  K9  	  R# 5i   Nr
   rI   .0rZ   r   s   & r   	<genexpr>.test_pack_unpack_with_numpy.<locals>.<genexpr>   +     @R3qww<3qww<!#33R   ?AFTc              3      <"   T F7  p\        VP                  4      \        SP                  4      ^,           8H  x  K9  	  R# 5irt   rv   rw   s   & r   ry   rz      r{   r|   c              3      <"   T F7  p\        VP                  4      \        SP                  4      ^,           8H  x  K9  	  R# 5irt   rv   rw   s   & r   ry   rz      r{   r|   )casesrI   r   rG   rH   r#   rV   rW   r   re   r   rN   )caserI   r   _lastrsr   s         @r   test_pack_unpack_with_numpyr      s`    

,,JJuQC!qc?G4]];''AaS1#:N0NO (]];''AaS1#8 (]];''AaS1#8 ( 	QC!rd+W5QC"s+W5RD1#s+W5$Q!qcB4(8'B
1zz$((*%***]];''1sQC"&0 (
 ]];''1taS1#&0 ( Q!Q!Q"a 97Cs@R@sss@R@@@@@Q!Q!R1a& 97Cs@R@sss@R@@@@@Q!R1a&1a& 97Cs@R@sss@R@@@@@ ]];''1AwaS)73 (]];''11vsQG,g6 ( ]];''11vsQG,g6 (]];''11v2w,g6 (]];''12wA,g6 ( 	QC!rd+W5QC"s+W5RD1#s+W5 	QGaV,g6s  ('''''''' (''
 ('' (''''' (''''''''sx   .R3R.1S&S$S**S>)T)T&)T:)UR+.R?SS'*S;>TT#&T7:UUc                     \        R R R7       EF|  p \        RV P                   24       Rp\         EFS  p\        pVP
                  pVP                  p\        P                  V4      pV P                  V4      pV! V^.^.^..V4       \        P                  ! \        4      ;_uu_ 4        \        V^.^.^..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        V^.^.^..V4       RRR4       V! V^.^.R..V4       V! V^.R.^..V4       V! VR.^.^..V4       \        P                  ! \        4      ;_uu_ 4        \        V^.^.R..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        VR.^.^..V4       RRR4       V! V^^.^^.R^..V4       V! V^^.^R.^^..V4       V! V^R.^^.^^..V4       \        P                  ! \        4      ;_uu_ 4        \        VR^.^.^..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        V^^.^.^R..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        V^^.^.^R..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        V^^.^R.^..V4       RRR4       \        P                  ! \        4      ;_uu_ 4        \        V^R.^^.^..V4       RRR4       V'       g   EK  V! V^.^.R..V4       V! V^ .^.R..V4       V! V^ .R.^..V4       V! VR.^.^ ..V4       V! V^R.^^..V4       EKV  	  EK  	  R#   + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     ELV; i  + '       g   i     EL,; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     ELp; i  + '       g   i     ELD; i  + '       g   i     EL; i)F)symboliclayersztest packing against numpy for TNr   )r   printframework_namer   r/   rI   r   rG   rH   
from_numpyrV   rW   r   r   )backendcheck_zero_lenr   r#   rI   r   r   s          r   test_pack_unpack_against_numpyr      s   (%FF/0F0F/GHIED;OJJEllG

5!A""1%AAaS1#8{++qA3aS/73 , {++qA3aS/73 , AaS2$/9AbTA3/9AqcA3/9 {++qA3bT*G4 ,
 {++qB4!qc*G4 , AAAQ8'BAAB!Q8'BAB!Q!Q8'B {++qB7QC!-w7 ,{++qAq6A3B0': , {++qAq6A3B0': ,{++qAq6Ar7QC0': ,{++qAr7QFQC0': , ~QC!rd#3W=QC!rd#3W=QC"s#3W=RD1#s#3W=  QGaV#4g>q 	 G ,++ ,++ ,++
 ,++ ,+++++ ,++++++++sl   ,N	*N%	N9	O	O!	O5	P		P	P1	N"%N69O
O!O25P	PP.1Qc            
         ^ RI p ^ RIHp V P                  R8  d   \        P
                  ! 4        \         EF  pVP                  pVP                  p\        P                  V4      pV P                  V4      p^.^.^..^.^.R..^.^.R^..^^.. ROR..3 F  p\        WWV4      pVP                  WgV4      p	\        W4       F9  w  r\        P                  ! WP!                  V^ ,           4      4      '       d   K9  Q h	  \#        W4      w  rVP#                  W4      w  rW8X  g   Q h\        P                  ! WP!                  V4      4      '       d   K  Q h	  ^^..^.^..^.^.R..^.^^..^.^.R^..3 F:  p\        P$                  ! \&        4      ;_uu_ 4        \        WWV4       RRR4       K<  	  EK  	  R#   + '       g   i     KV  ; i)r   N)	array_apiz2.0.0r   )ru   ru   ru   )numpyeinopsr   __version__rV   skipr   rI   r   rG   rH   from_dlpackr   r   r   r   r   r   rW   r   )xpAAr   rI   r   r+   x_xpr   
x_np_split
x_xp_splitr   r   x_agg_npps1x_agg_xpr"   s                   r   test_pack_unpack_array_apir     s   &	~~

,,zz% ~~d# S1#sOS1#tS1#AwVY%	
B  '2J4W5JJ3{{1jjQ&78888 4 !5MHGGJ8MH::;;xH)=>>>>
" VHS1#JS1#tS1a&MS1#Aw
B {++t) ,+
- : ,++s   !G	G)      )rh   )rh      )r   rh   )rh         )r   rh   r   )r   r   rh   )dataclassesrl   r   r   rV   r   r   r   r   r   einops.testsr   rH   default_rngrG   r   r#   r/   r&   r^   	dataclassr`   r   r   r   r   rp   r   r   <module>r      s        5 5 .ii3.4 8)v + + + 469%69%9n-9n-9n-		<7~=?@&*r   