
    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                   x    \ 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r
g)
StringSplit   Nc                  f   [         R                  R                  SS/SS/SS S9n [        R                  " SS/5      R                  [        5      n[        R                  " S	S
/SS//5      R                  [        5      n[        R                  " SS/[        R                  S9n[        U U/X#/SS9  g )Nr   x
substringslength.inputsoutputs	delimitermaxsplitzabc.comdef.netabccomdefnet   dtypetest_string_split_basicr   r   name	onnxhelper	make_nodenparrayastypeobjectint64r   noder	   r
   r   s       b/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/string_split.pyexport_basicStringSplit.export_basic   s    {{$$5!8, % 
 HHi+,33F;XXu~u~>?FFvN
1a&13(*		
    c                     [         R                  R                  SS/SS/SS9n [        R                  " SS/S	S
//5      R                  [        5      n[        R                  " / SQ/ SQ// SQ/ SQ//5      R                  [        5      n[        R                  " SS/SS//[        R                  5      n[        U U/X#/SS9  g )Nr   r	   r
   r   r   )r   r   r   zhello worldr   zo n n xzthe quick brown fox)helloworld )r   r/   r/   )onzn x)thequickz	brown fox      test_string_split_maxsplitr   r   r&   s       r(   export_maxsplitStringSplit.export_maxsplit$   s    {{$$5!8,	 % 
 HHY')5J)KL

&. 	
 XX')<="$AB

 &. 	 Aq6Aq6*BHH53(-		
r+   c                  f   [         R                  R                  SS/SS/SS S9n [        R                  " SS/5      R                  [        5      n[        R                  " / S	Q/ S
Q/5      R                  [        5      n[        R                  " SS/[        R                  S9n[        U U/X#/SS9  g )Nr   r	   r
   r   -r   z	o-n-n--x-z	o-n----nx)r0   r1   r1   r/   r	   r/   )r0   r1   r/   r/   r/   nx   r   (test_string_split_consecutive_delimitersr   r   r&   s       r(   export_consecutive_delimiters)StringSplit.export_consecutive_delimitersA   s    {{$$5!8, % 
 HHk;/077?XX)+GH

&. 	 1a&13(;		
r+   c            	         S H  u  p[         R                  R                  SS/SS/U S S9n[        R                  " / SQ5      R                  [        5      n[        R                  " / SQ/ SQ/ SQ/5      R                  [        5      n[        R                  " / S	Q[        R                  S
9n[        UU/XE/US9  M     g )N))r/   (test_string_split_empty_string_delimiter)Ntest_string_split_no_delimiterr   r	   r
   r   r   )zhello world !z  hello   world !z hello world   ! )r-   r.   !)r5   r5   r5   r   r   r   )r   	test_namer'   r	   r
   r   s         r(   export_empty_string_delimiter)StringSplit.export_empty_string_delimiterZ   s    %
 I ;;((u%x0# ) D KfVn  +++ fVn  XXirxx8Fs#,	5%
r+   c            
         [         R                  R                  SS/SS/S S S9n [        R                  " / 5      R                  [        5      n[        R                  " / 5      R                  [        5      R                  SS5      n[        R                  " / [        R                  S9n[        U U/X#/S[         R                  R                  [         R                  R                  S	5      S /S
9  g )Nr   r	   r
   r   r   r   r   test_string_split_empty_tensor)r   N)r   r   r   output_type_protos)r   r   r    r!   r"   r#   r$   reshaper%   r   make_tensor_type_protoTensorProtoSTRINGr&   s       r(   export_empty_string_split%StringSplit.export_empty_string_split}   s    {{$$5!8, % 
 HHRL'XXb\((088A>
"BHH-3(12243C3C3J3JIV 		
r+    )returnN)__name__
__module____qualname____firstlineno__staticmethodr)   r7   r>   rE   rN   __static_attributes__rP   r+   r(   r   r      se    
 
, 
 
8 
 
0    D 
 
r+   r   )numpyr!   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rP   r+   r(   <module>r[      s!   
   , .J
$ J
r+   