
    /j                    V    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	dS )    )annotationsN)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returnNonec                 v   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
           t          j                            ddd          d	z                      t          j
                  }t          j                            ddd          d	z                      t          j
                  }t          j        ||          }t          | ||g|gd           t          j                            ddd          d	z                      t          j                  }t          j                            ddd          d	z                      t          j                  }t          j        ||          }t          | ||g|gd           t          j                            ddt          j                  }t          j                            ddt          j                  }t          j        ||          }t          | ||g|gd           t          j                            ddt          j                  }t          j                            ddt          j                  }t          j        ||          }t          | ||g|gd           t          j                            ddt          j                  }t          j                            ddt          j                  }t          j        ||          }t          | ||g|gd           t          j                            ddt          j                  }t          j                            ddt          j                  }t          j        ||          }t          | ||g|gd           d S )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       f/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/equal.pyexportzEqual.export   sg   {$$:E % 
 
 Y__Q1%%*2228<<Y__Q1%%*2228<<HQNNtQFQClCCCCY__Q1%%*2227;;Y__Q1%%*2227;;HQNNtQFQC6GHHHHY__Q1%%*2228<<Y__Q1%%*2228<<HQNNtQFQC6HIIIIIbyAAIbyAAHQNNtQFQC6HIIIIIby	BBIby	BBHQNNtQFQC6IJJJJIby	BBIby	BBHQNNtQFQC6IJJJJIby	BBIby	BBHQNNtQFQC6IJJJJJJ    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"   r#   r$   r%   r&   r'   r(   r)   r   r1   s       r3   export_equal_broadcastzEqual.export_equal_broadcast9   s    {$$:E % 
 
 Y__Q1%%*2228<<Y__Q"$,,RX66HQNNtQFQC6HIIIIIIr5   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r   string3test_equal_stringr   	r!   r"   r#   r$   arrayr   objectr)   r   r1   s       r3   export_equal_stringzEqual.export_equal_stringF   s    {$$:E % 
 

 Hi+28F3C3CDDDHi+28F3C3CDDDHQNNtQFQC6IJJJJJJr5   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?   r1   s       r3   export_equal_string_broadcastz#Equal.export_equal_string_broadcastR   s    {$$:E % 
 

 Hi+28F3C3CDDDHi[(8(8999HQNNtQFQC6STTTTTTr5   N)r   r	   )__name__
__module____qualname__staticmethodr4   r8   rB   rE    r5   r3   r   r      s        (K (K (K \(KT 
J 
J 
J \
J 	K 	K 	K \	K 	U 	U 	U \	U 	U 	Ur5   r   )

__future__r   numpyr$   r!   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rJ   r5   r3   <module>rO      s    # " " " " "      , , , , , , . . . . . .OU OU OU OU OUD OU OU OU OU OUr5   