
    h".                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                       \ rS rSr\SS j5       r\SS j5       r\SS j5       r\SS j5       r\SS j5       r	\SS j5       r
\SS	 j5       r\SS
 j5       r\SS j5       r\SS j5       rSrg)Split   Nc            
      *   [         R                  " / SQ5      R                  [         R                  5      n [        R
                  R                  SS// SQSS9n[         R                  " SS/5      R                  [         R                  5      [         R                  " S	S
/5      R                  [         R                  5      [         R                  " SS/5      R                  [         R                  5      /n[        UU /US[        R
                  R                  SS5      /S9  [         R                  " SS/5      R                  [         R                  5      n[        R
                  R                  SSS/SS/SS9n[         R                  " SS/5      R                  [         R                  5      [         R                  " / SQ5      R                  [         R                  5      /n[        UX/US[        R
                  R                  SS5      /S9  g )N      ?       @      @      @      @      @r   inputoutput_1output_2output_3r   inputsoutputsaxisr
   r   r   r   r   r   !test_split_equal_parts_1d_opset13    r   r   nameopset_imports      splitr   r   r   r   r   r   $test_split_variable_parts_1d_opset13
nparrayastypefloat32onnxhelper	make_noder   make_opsetidint64
node_inputnodeexpected_outputsr!   s       [/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/split.pyexport_1d_opset13Split.export_1d_opset13   s   XX<=DDRZZP
{{$$98	 % 
 HHc3Z ''

3HHc3Z ''

3HHc3Z ''

3

 	<$4;;33B;<	
 !Q ''1{{$$W%,	 % 
 HHc3Z ''

3HH)*11"**=
 	&$7;;33B;<	
    c            
         [         R                  " / SQ/ SQ/5      R                  [         R                  5      n [        R
                  R                  SS/SS/SS9n[         R                  " / S	Q/ S
Q/5      R                  [         R                  5      [         R                  " / SQ/ SQ/5      R                  [         R                  5      /n[        UU /US[        R
                  R                  SS5      /S9  [         R                  " SS/5      R                  [         R                  5      n[        R
                  R                  SSS/SS/SS9n[         R                  " SS/SS//5      R                  [         R                  5      [         R                  " / SQ/ SQ/5      R                  [         R                  5      /n[        UX/US[        R
                  R                  SS5      /S9  g )Nr	         @       @      "@      $@      &@      (@r   r   r   r      r   r
   r   r   r8   r9   r:   r   r   r   r;   r<   r=   !test_split_equal_parts_2d_opset13r   r   r   r   r    r!   r
   r   r8   r9   r"   r:   r;   r<   r=   $test_split_variable_parts_2d_opset13r$   r.   s       r2   export_2d_opset13Split.export_2d_opset139   s   XX+-NO

&
 	 {{$$WI
J/Ga % 

 HHo78??

KHHo'9:;BB2::N

 	<$4;;33B;<	
 !Q ''1{{$$W%,	 % 
 HHsCj3*-.55bjjAHH*,CDELL


 	&$7;;33B;<	
r5   c            
      &   [         R                  " / SQ5      R                  [         R                  5      n [        R
                  R                  SS// SQS9n[         R                  " SS/5      R                  [         R                  5      [         R                  " SS	/5      R                  [         R                  5      [         R                  " S
S/5      R                  [         R                  5      /n[        UU /US[        R
                  R                  SS5      /S9  [         R                  " SS/5      R                  [         R                  5      n[        R
                  R                  SSS/SS/S9n[         R                  " SS/5      R                  [         R                  5      [         R                  " / SQ5      R                  [         R                  5      /n[        UX/US[        R
                  R                  SS5      /S9  g )Nr	   r   r   r   r   r   r
   r   r   r   r   r   +test_split_equal_parts_default_axis_opset13r   r   r   r   r    r!   r   r   r"   .test_split_variable_parts_default_axis_opset13r$   r.   s       r2   export_default_values_opset13#Split.export_default_values_opset13g   s   XX<=DDRZZP
 {{$$WI/S % 

 HHc3Z ''

3HHc3Z ''

3HHc3Z ''

3

 	<$>;;33B;<	
 !Q ''1{{$$Wg.Z8P % 

 HHc3Z ''

3HH)*11"**=
 	&$A;;33B;<	
r5   c            
         [         R                  " / 5      R                  [         R                  5      n [         R                  " / SQ5      R                  [         R                  5      n[
        R                  R                  SSS// SQS9n[         R                  " / 5      R                  [         R                  5      [         R                  " / 5      R                  [         R                  5      [         R                  " / 5      R                  [         R                  5      /n[        UX/US[
        R                  R                  SS	5      /S
9  g )Nr   r   r   r   r   r!   r   rI   #test_split_zero_size_splits_opset13r   r   r   )
r%   r&   r'   r(   r-   r)   r*   r+   r   r,   r/   r!   r0   r1   s       r2   export_zero_size_splits_opset13%Split.export_zero_size_splits_opset13   s     XXb\((4
 #**2884{{$$W%8 % 
 HHRL

+HHRL

+HHRL

+

 	&$6;;33B;<	
r5   c                     [         R                  " / SQ5      R                  [         R                  5      n [        R
                  R                  SS// SQSSS9n[         R                  " SS	/5      R                  [         R                  5      [         R                  " S
S/5      R                  [         R                  5      [         R                  " SS/5      R                  [         R                  5      /n[        UU /USS9  [         R                  " SS/5      R                  [         R                  5      n[        R
                  R                  SSS/SS/SS9n[         R                  " SS	/5      R                  [         R                  5      [         R                  " / SQ5      R                  [         R                  5      /n[        UX/USS9  g )Nr	   r   r   r   r      r   r   r   num_outputsr
   r   r   r   r   r   !test_split_equal_parts_1d_opset18r   r   r   r   r    r!   r   r   r   r"   $test_split_variable_parts_1d_opset18	r%   r&   r'   r(   r)   r*   r+   r   r-   r.   s       r2   export_1d_opset18Split.export_1d_opset18   sy   XX<=DDRZZP
{{$$98 % 
 HHc3Z ''

3HHc3Z ''

3HHc3Z ''

3

 	<$4		
 !Q ''1{{$$W%,	 % 
 HHc3Z ''

3HH)*11"**=
 	&$7		
r5   c                  l   [         R                  " / SQ/ SQ/5      R                  [         R                  5      n [        R
                  R                  SS/SS/SSS	9n[         R                  " / S
Q/ SQ/5      R                  [         R                  5      [         R                  " / SQ/ SQ/5      R                  [         R                  5      /n[        UU /USS9  [         R                  " SS/5      R                  [         R                  5      n[        R
                  R                  SSS/SS/SS9n[         R                  " SS/SS//5      R                  [         R                  5      [         R                  " / SQ/ SQ/5      R                  [         R                  5      /n[        UX/USS9  g )Nr	   r7   r   r   r   r   r>   r   rV   r?   r@   rA   rB   test_split_equal_parts_2drY   r    r!   r   r
   r   r8   r9   r"   rD   $test_split_variable_parts_2d_opset18r[   r.   s       r2   export_2d_opset18Split.export_2d_opset18   s   XX+-NO

&
 	 {{$$9, % 
 HHo78??

KHHo'9:;BB2::N

 	<$,		
 !Q ''1{{$$W%,	 % 
 HHsCj3*-.55bjjAHH*,CDELL


 	&$7		
r5   c                     [         R                  " / SQ5      R                  [         R                  5      n [        R
                  R                  SS// SQSS9n[         R                  " SS/5      R                  [         R                  5      [         R                  " S	S
/5      R                  [         R                  5      [         R                  " SS/5      R                  [         R                  5      /n[        UU /USS9  [         R                  " SS/5      R                  [         R                  5      n[        R
                  R                  SSS/SS/S9n[         R                  " SS/5      R                  [         R                  5      [         R                  " / SQ5      R                  [         R                  5      /n[        UX/USS9  g )Nr	   r   r   r   rU   r   r   rW   r
   r   r   r   r   r   +test_split_equal_parts_default_axis_opset18rY   r   r    r!   r   r   rI   r"   .test_split_variable_parts_default_axis_opset18r[   r.   s       r2   export_default_values_opset18#Split.export_default_values_opset18  ss   XX<=DDRZZP
 {{$$98	 % 
 HHc3Z ''

3HHc3Z ''

3HHc3Z ''

3

 	<$>		
 !Q ''1{{$$Wg.Z8P % 

 HHc3Z ''

3HH)*11"**=
 	&$A		
r5   c                  d   [         R                  " / 5      R                  [         R                  5      n [         R                  " / SQ5      R                  [         R                  5      n[
        R                  R                  SSS// SQS9n[         R                  " / 5      R                  [         R                  5      [         R                  " / 5      R                  [         R                  5      [         R                  " / 5      R                  [         R                  5      /n[        UX/USS9  g )	NrO   r   r   r!   r   rI   #test_split_zero_size_splits_opset18rY   )	r%   r&   r'   r(   r-   r)   r*   r+   r   rQ   s       r2   export_zero_size_splits_opset18%Split.export_zero_size_splits_opset18+  s     XXb\((4
 #**2884{{$$W%8 % 
 HHRL

+HHRL

+HHRL

+

 	&$6		
r5   c                  p   [         R                  " / SQ5      R                  [         R                  5      n [        R
                  R                  SS// SQSS9n[         R                  " SS/5      R                  [         R                  5      [         R                  " S	S
/5      R                  [         R                  5      [         R                  " SS/5      R                  [         R                  5      [         R                  " S/5      R                  [         R                  5      /n[        UU /USS9  g )N)r
   r   r   r   r   r   r8   r   r   )r   r   r   output_4r    rd   r
   r   r   r   r   r   r8   "test_split_1d_uneven_split_opset18rY   r%   r&   r'   r(   r)   r*   r+   r   r/   r0   r1   s      r2   export_1d_uneven_split_opset18$Split.export_1d_uneven_split_opset18D  s    XXABII"**U
 {{$$9D	 % 
 HHc3Z ''

3HHc3Z ''

3HHc3Z ''

3HHcUO""2::.	
 	<$5		
r5   c                  ,   [         R                  " / SQ/ SQ/5      R                  [         R                  5      n [        R
                  R                  SS// SQSSS9n[         R                  " / S	Q/ S
Q/5      R                  [         R                  5      [         R                  " / SQ/ SQ/5      R                  [         R                  5      [         R                  " SS/SS//5      R                  [         R                  5      /n[        UU /USS9  g )N)r
   r   r   r   r   r   r8   r9   )r:   r;   r<   r=         *@      ,@      .@      0@r   r   r   r>   rU   rV   r?   )r:   r;   r<   rA   )r=   ru   rv   r8   r9   rw   rx   "test_split_2d_uneven_split_opset18rY   rp   rq   s      r2   export_2d_uneven_split_opset18$Split.export_2d_uneven_split_opset18]  s    XX8?

 &
 	 {{$$98 % 
 HHo'89:AA"**MHHo'9:;BB2::NHHsCj4,/077

C
 	<$5		
r5    )returnN)__name__
__module____qualname____firstlineno__staticmethodr3   rF   rL   rR   r\   ra   rg   rk   rr   rz   __static_attributes__r|   r5   r2   r   r      s    )
 )
V +
 +
Z $
 $
L 
 
2 (
 (
T -
 -
^ %
 %
N 
 
0 
 
0 
 
r5   r   )numpyr%   r)   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r|   r5   r2   <module>r      s!   
   , .m
D m
r5   