U
    rh}                     @  sH   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 deZ	dS )    )annotationsN)Base)expectc                   @  sT   e Zd ZeddddZeddddZeddddZeddd	d
ZdS )EqualNone)returnc                  C  sv   t jjdddgdgd} tjdddd	 tj}tjdddd	 tj}t||}t	| ||g|gd
d d S )Nr   xyzinputsoutputs         
   Z
test_equalr   r   name
onnxhelper	make_nodenprandomZrandnZastypeZint32equalr   noder   r	   r
    r   E/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/equal.pyexport   s    zEqual.exportc                  C  sr   t jjdddgdgd} tjdddd	 tj}tjdd	 tj}t||}t	| ||g|gd
d d S )Nr   r   r	   r
   r   r   r   r   r   Ztest_equal_bcastr   r   r   r   r   r   export_equal_broadcast   s    zEqual.export_equal_broadcastc                  C  sn   t jjdddgdgd} tjddgttd}tjdd	gttd}t||}t| ||g|gd
d d S )Nr   r   r	   r
   r   string1string2dtypeZstring3Ztest_equal_stringr   	r   r   r   r   arrayr$   objectr   r   r   r   r   r   export_equal_string(   s    zEqual.export_equal_stringc                  C  sl   t jjdddgdgd} tjddgttd}tjdgttd}t||}t| ||g|gd	d
 d S )Nr   r   r	   r
   r   r!   r"   r#   Ztest_equal_string_broadcastr   r%   r   r   r   r   export_equal_string_broadcast4   s    z#Equal.export_equal_string_broadcastN)__name__
__module____qualname__staticmethodr   r    r(   r)   r   r   r   r   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r   r   r   r   <module>   s
   