
    Si                    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)    )annotationsN)Base)expectc                  d    \ 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r	g)	Equal   c                    [         R                  R                  SSS/S/S9n [        R                  R                  SSS5      S	-  R                  [        R                  5      n[        R                  R                  SSS5      S	-  R                  [        R                  5      n[        R                  " X5      n[        XU/U/S
S9  [        R                  R                  SSS5      S	-  R                  [        R                  5      n[        R                  R                  SSS5      S	-  R                  [        R                  5      n[        R                  " X5      n[        XU/U/SS9  [        R                  R                  SSS5      S	-  R                  [        R                  5      n[        R                  R                  SSS5      S	-  R                  [        R                  5      n[        R                  " X5      n[        XU/U/SS9  [        R                  R                  SS[        R                  S9n[        R                  R                  SS[        R                  S9n[        R                  " X5      n[        XU/U/SS9  [        R                  R                  SS[        R                  S9n[        R                  R                  SS[        R                  S9n[        R                  " X5      n[        XU/U/SS9  [        R                  R                  SS[        R                  S9n[        R                  R                  SS[        R                  S9n[        R                  " X5      n[        XU/U/SS9  [        R                  R                  SS[        R                   S9n[        R                  R                  SS[        R                   S9n[        R                  " X5      n[        XU/U/SS9  g )Nr   xyzinputsoutputs         
   
test_equalr   r   nametest_equal_int8test_equal_int16   )r   r   r   )sizedtypetest_equal_uint8test_equal_uint16test_equal_uint32test_equal_uint64)onnxhelper	make_nodenprandomrandnastypeint32equalr   int8int16randintuint8uint16uint32uint64noder
   r   r   s       `/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/equal.pyexportEqual.export   s   {{$$:E % 
 YY__Q1%*22288<YY__Q1%*22288<HHQNtFQClCYY__Q1%*22277;YY__Q1%*22277;HHQNtFQC6GHYY__Q1%*22288<YY__Q1%*22288<HHQNtFQC6HIIIbyAIIbyAHHQNtFQC6HIIIby		BIIby		BHHQNtFQC6IJIIby		BIIby		BHHQNtFQC6IJIIby		BIIby		BHHQNtFQC6IJ    c                    [         R                  R                  SSS/S/S9n [        R                  R                  SSS5      S	-  R                  [        R                  5      n[        R                  R                  S5      S	-  R                  [        R                  5      n[        R                  " X5      n[        XU/U/S
S9  g )Nr   r
   r   r   r   r   r   r   r   test_equal_bcastr   )
r    r!   r"   r#   r$   r%   r&   r'   r(   r   r0   s       r2   export_equal_broadcastEqual.export_equal_broadcast9   s    {{$$:E % 
 YY__Q1%*22288<YY__Q"$,,RXX6HHQNtFQC6HIr5   c                 L   [         R                  R                  SSS/S/S9n [        R                  " SS/[        R
                  " [        5      S9n[        R                  " SS	/[        R
                  " [        5      S9n[        R                  " X5      n[        XU/U/S
S9  g )Nr   r
   r   r   r   string1string2r   string3test_equal_stringr   	r    r!   r"   r#   arrayr   objectr(   r   r0   s       r2   export_equal_stringEqual.export_equal_stringF   s    {{$$:E % 

 HHi+288F3CDHHi+288F3CDHHQNtFQC6IJr5   c                 J   [         R                  R                  SSS/S/S9n [        R                  " SS/[        R
                  " [        5      S9n[        R                  " S/[        R
                  " [        5      S9n[        R                  " X5      n[        XU/U/S	S
9  g )Nr   r
   r   r   r   r;   r<   r=   test_equal_string_broadcastr   r@   r0   s       r2   export_equal_string_broadcast#Equal.export_equal_string_broadcastR   s    {{$$:E % 

 HHi+288F3CDHHi[(89HHQNtFQC6STr5    N)returnNone)
__name__
__module____qualname____firstlineno__staticmethodr3   r8   rC   rG   __static_attributes__rI   r5   r2   r   r      sY    (K (KT 
J 
J 	K 	K 	U 	Ur5   r   )

__future__r   numpyr#   r    onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rI   r5   r2   <module>rV      s&    #   , .OUD OUr5   