
    hZ                     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                   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   Nc                     [         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  g )Nr   xyzinputsoutputs         
   
test_equalr   r   name
onnxhelper	make_nodenprandomrandnastypeint32equalr   noder	   r
   r   s       [/var/www/fran/franai/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    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    s       r"   export_equal_broadcastEqual.export_equal_broadcast   s    {{$$:E % 
 YY__Q1%*22288<YY__Q"$,,RXX6HHQNtFQC6HIr%   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dtypestring3test_equal_stringr   	r   r   r   r   arrayr.   objectr   r   r    s       r"   export_equal_stringEqual.export_equal_string'   s    {{$$:E % 

 HHi+288F3CDHHi+288F3CDHHQNtFQC6IJr%   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   r1   r    s       r"   export_equal_string_broadcast#Equal.export_equal_string_broadcast3   s    {{$$:E % 

 HHi+288F3CDHHi[(89HHQNtFQC6STr%    )returnN)
__name__
__module____qualname____firstlineno__staticmethodr#   r(   r4   r8   __static_attributes__r:   r%   r"   r   r      sX    
D 
D 
J 
J 	K 	K 	U 	Ur%   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r:   r%   r"   <module>rE      s!   
   , .1UD 1Ur%   