
    h                     N    S SK Jr  S SKrS SKrS SKJr  S SKJr   " S S\5      r	g)    )SequenceN)Base)expectc                   (    \ rS rSr\SS j5       rSrg)NormalizeStrings   Nc            	      
   S[         R                  R                  S[        [           S[        [           S[         R                  R
                  4S jn S/n[         R                  R                  SS/S	/S
US9n[        R                  " / SQ5      R                  [        5      n[        R                  " / SQ5      R                  [        5      nU " US/S/5      n[         R                  R                  US[         R                  R                  SS5      /S9n[        UU/U/SS9  [         R                  R                  SS/S	/S
S9n[        R                  " SS/5      R                  [        5      nUnU " US/S/5      n[         R                  R                  US[         R                  R                  SS5      /S9n[        UU/U/SS9  S/n[         R                  R                  SS/S	/SS
US9n[        R                  " / SQ5      R                  [        5      n[        R                  " / SQ5      R                  [        5      nU " US/S/5      n[         R                  R                  US[         R                  R                  SS5      /S9n[        UU/U/SS9  S/n[         R                  R                  SS/S	/SS
US9n[        R                  " / SQ5      R                  [        5      n[        R                  " / SQ5      R                  [        5      nU " US/S/5      n[         R                  R                  US[         R                  R                  SS5      /S9n[        UU/U/SS9  S/n[         R                  R                  SS/S	/SUS 9nS
S!/nS
S/n[        R                  " / S"Q5      R                  [        5      R                  U5      n[        R                  " / S#Q5      R                  [        5      R                  U5      nU " X'U5      n[         R                  R                  US[         R                  R                  SS5      /S9n[        UU/U/S$S9  S/n[         R                  R                  SS/S	/SS%US9n[        R                  " SS/5      R                  [        5      n[        R                  " S/5      R                  [        5      nU " US/S
/5      n[         R                  R                  US[         R                  R                  SS5      /S9n[        UU/U/S&S9  g )'Nnodeinput_shapeoutput_shapereturnc           
      (   [         R                  R                  U /S[         R                  R                  S[         R                  R
                  U5      /[         R                  R                  S[         R                  R
                  U5      /S9nU$ )NStringNormalizerxy)nodesnameinputsoutputs)onnxhelper
make_graphmake_tensor_value_infoTensorProtoSTRING)r
   r   r   graphs       g/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/model/stringnormalizer.pyr   +NormalizeStrings.export.<locals>.make_graph   s    
 KK**f'KK66T--44k KK66T--44l + E L    mondayr   r   r      )r   r   is_case_sensitive	stopwords)r    tuesday	wednesdaythursday)r$   r%   r&         zbackend-test 
   )producer_nameopset_imports5test_strnorm_model_monday_casesensintive_nochangecase)r   r   r   )r   r   r"   r$      +test_strnorm_model_nostopwords_nochangecaseLOWER)r   r   case_change_actionr"   r#   .test_strnorm_model_monday_casesensintive_lowerUPPER)TUESDAY	WEDNESDAYTHURSDAY.test_strnorm_model_monday_casesensintive_upper)r   r   r1   r#      )Mondayr$   r%   r9   r$   r%   )r4   r5   r4   r5   3test_strnorm_model_monday_insensintive_upper_twodimr   &test_strnorm_model_monday_empty_output)r   r   	NodeProtor   int
GraphProto	make_nodenparrayastypeobjectmake_model_gen_versionmake_opsetidr   reshape)	r   r#   r
   r   r   r   modelr   r   s	            r   exportNormalizeStrings.export   s.   	++''	!#	 #3-	 [[##		, J	{{$$5E % 
 HHCDKKFSHH9:AA&I4!qc*22(;;33B;< 3 

 	3CH		
 {{$$useq % 
 HHh	*+226:4!qc*22(;;33B;< 3 

 	3C>		
 J	{{$$5E& % 
 HHCDKKFSHH9:AA&I4!qc*22(;;33B;< 3 

 	3CA		
 J	{{$$5E& % 
 HHCDKKFSHH9:AA&I4!qc*22(;;33B;< 3 

 	3CA		
 J	{{$$5E& % 
 !f1vHHT VF^W[! 	
 HHEFVF^W\" 	
 4l;22(;;33B;< 3 

 	3CF		
 J	{{$$5E& % 
 HHh)*11&9HHbTN!!&)4!qc*22(;;33B;< 3 

 	3C9		
r    )r   N)__name__
__module____qualname____firstlineno__staticmethodrH   __static_attributes__rJ   r   r   r   r      s    z
 z
r   r   )
typingr   numpyr@   r   onnx.backend.test.case.baser   onnx.backend.test.case.modelr   r   rJ   r   r   <module>rU      s$       , /|
t |
r   