
    bIg}                    N    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e      Z	y)    )annotationsN)Base)expectc                  T    e Zd Zedd       Zedd       Zedd       Zedd       Zy)Equalc                    t         j                  j                  dddgdg      } t        j                  j                  ddd      d	z  j                  t        j                        }t        j                  j                  ddd      d	z  j                  t        j                        }t        j                  ||      }t        | ||g|gd
       y )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/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/equal.pyexportzEqual.export   s    {{$$:E % 
 YY__Q1%*22288<YY__Q1%*22288<HHQNtQFQClC    c                    t         j                  j                  dddgdg      } t        j                  j                  ddd      d	z  j                  t        j                        }t        j                  j                  d      d	z  j                  t        j                        }t        j                  ||      }t        | ||g|gd
       y )Nr   r	   r
   r   r   r   r   r   r   test_equal_bcastr   r   r    s       r"   export_equal_broadcastzEqual.export_equal_broadcast   s    {{$$:E % 
 YY__Q1%*22288<YY__Q"$,,RXX6HHQNtQFQC6HIr$   c                 ^   t         j                  j                  dddgdg      } t        j                  ddgt        j
                  t                    }t        j                  dd	gt        j
                  t                    }t        j                  ||      }t        | ||g|gd
       y )Nr   r	   r
   r   r   string1string2dtypestring3test_equal_stringr   	r   r   r   r   arrayr,   objectr   r   r    s       r"   export_equal_stringzEqual.export_equal_string(   s    {{$$:E % 

 HHi+288F3CDHHi+288F3CDHHQNtQFQC6IJr$   c                 \   t         j                  j                  dddgdg      } t        j                  ddgt        j
                  t                    }t        j                  dgt        j
                  t                    }t        j                  ||      }t        | ||g|gd	
       y )Nr   r	   r
   r   r   r)   r*   r+   test_equal_string_broadcastr   r/   r    s       r"   export_equal_string_broadcastz#Equal.export_equal_string_broadcast4   s    {{$$:E % 

 HHi+288F3CDHHi[(89HHQNtQFQC6STr$   N)returnNone)__name__
__module____qualname__staticmethodr#   r'   r2   r5    r$   r"   r   r      sX    
D 
D 
J 
J 	K 	K 	U 	Ur$   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r<   r$   r"   <module>rA      s$    #   , .1UD 1Ur$   