
    NgZ                     J    d dl Zd dlZd dlmZ d dlmZ  G d de          ZdS )    N)Base)expectc                   n    e Zd Zedd            Zedd            Zedd            Zedd            ZdS )EqualreturnNc                     t           j                            dddgdg          } t          j                            ddd          d	z                      t          j                  }t          j                            ddd          d	z                      t          j                  }t          j        ||          }t          | ||g|gd
           d S )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/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/equal.pyexportzEqual.export   s    {$$:E % 
 
 Y__Q1%%*2228<<Y__Q1%%*2228<<HQNNtQFQClCCCCCC    c                     t           j                            dddgdg          } t          j                            ddd          d	z                      t          j                  }t          j                            d          d	z                      t          j                  }t          j        ||          }t          | ||g|gd
           d S )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 % 
 
 Y__Q1%%*2228<<Y__Q"$,,RX66HQNNtQFQC6HIIIIIIr$   c                  b   t           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
           d S )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 % 
 

 Hi+28F3C3CDDDHi+28F3C3CDDDHQNNtQFQC6IJJJJJJr$   c                  `   t           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	
           d S )Nr   r	   r
   r   r   r)   r*   r+   test_equal_string_broadcastr   r/   r    s       r"   export_equal_string_broadcastz#Equal.export_equal_string_broadcast3   s    {$$:E % 
 

 Hi+28F3C3CDDDHi[(8(8999HQNNtQFQC6STTTTTTr$   )r   N)__name__
__module____qualname__staticmethodr#   r'   r2   r5    r$   r"   r   r      s        
D 
D 
D \
D 
J 
J 
J \
J 	K 	K 	K \	K 	U 	U 	U \	U 	U 	Ur$   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r:   r$   r"   <module>r>      s~   
      , , , , , , . . . . . .1U 1U 1U 1U 1UD 1U 1U 1U 1U 1Ur$   