o
    soiL                     @  sH   d dl mZ d dlZd dlZd dlmZ d dlmZ G dd deZ	dS )    )annotationsN)Base)expectc                   @  s`   e Zd ZedddZedddZedddZedd	d
ZedddZedddZ	dS )StringNormalizerreturnNonec                  C  sH   t ddgt} | }tjjddgdgdd}t|| g|gdd	 d S )
Nmondaytuesdayr   xy   )inputsoutputsis_case_sensitive+test_strnormalizer_nostopwords_nochangecaser   r   namenparrayastypeobjectonnxhelper	make_noder   )inputoutputnode r   `/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/stringnormalizer.pyexport_nostopwords_nochangecase   s   
z0StringNormalizer.export_nostopwords_nochangecasec                  C  s`   t g dt} t g dt}dg}tjjddgdgd|d}t|| g|gd	d
 d S )Nr   r	   	wednesdaythursdayr	   r"   r#   r   r   r
   r   r   )r   r   r   	stopwords<test_strnormalizer_export_monday_casesensintive_nochangecaser   r   r   r   r%   r   r   r   r   )export_monday_casesensintive_nochangecase!   s    
z:StringNormalizer.export_monday_casesensintive_nochangecasec                  C  b   t g dt} t g dt}dg}tjjddgdgdd|d	}t|| g|gd
d d S )Nr!   r$   r   r   r
   r   LOWERr   r   r   case_change_actionr   r%   5test_strnormalizer_export_monday_casesensintive_lowerr   r   r'   r   r   r   "export_monday_casesensintive_lower5   "   
z3StringNormalizer.export_monday_casesensintive_lowerc                  C  r)   )Nr!   )TUESDAY	WEDNESDAYTHURSDAYr   r   r
   r   UPPERr   r+   5test_strnormalizer_export_monday_casesensintive_upperr   r   r'   r   r   r   "export_monday_casesensintive_upperJ   r/   z3StringNormalizer.export_monday_casesensintive_upperc                  C  s`   t ddgt} t dgt}dg}tjjddgdgdd|d}t|| g|gd	d
 d S )Nr    r   r
   r   r3   r   r+   -test_strnormalizer_export_monday_empty_outputr   r   r'   r   r   r   export_monday_empty_output_   s"   
z+StringNormalizer.export_monday_empty_outputc                  C  st   t g dtddg} t g dtddg}dg}tjjddgd	gd
|d}t|| g|gdd d S )N)Mondayr	   r"   r9   r	   r"   r      )r0   r1   r0   r1      r   r   r
   r   r3   )r   r   r,   r%   :test_strnormalizer_export_monday_insensintive_upper_twodimr   )	r   r   r   r   reshaper   r   r   r   r'   r   r   r   'export_monday_insensintive_upper_twodimt   s0   


z8StringNormalizer.export_monday_insensintive_upper_twodimN)r   r   )
__name__
__module____qualname__staticmethodr    r(   r.   r5   r8   r>   r   r   r   r   r      s    r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r   r   r   <module>   s   