o
    wi                     @  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                   @  sD   e Zd ZedddZedddZedddZedd	d
ZdS )EqualreturnNonec                  C  sR  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 tjdddd	 tj
}tjdddd	 tj
}t||}t	| ||g|gdd tjdddd	 tj}tjdddd	 tj}t||}t	| ||g|gdd tjjddtjd}tjjddtjd}t||}t	| ||g|gdd tjjddtjd}tjjddtjd}t||}t	| ||g|gdd tjjddtjd}tjjddtjd}t||}t	| ||g|gdd tjjddtjd}tjjddtjd}t||}t	| ||g|gd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
    r0   ^/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/equal.pyexport   sB   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   test_equal_bcastr   )
r   r   r    r!   r"   r#   r$   r%   r&   r   r.   r0   r0   r1   export_equal_broadcast9   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r   string3test_equal_stringr   	r   r   r    r!   arrayr   objectr&   r   r.   r0   r0   r1   export_equal_stringF   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   r5   r6   r7   test_equal_string_broadcastr   r:   r.   r0   r0   r1   export_equal_string_broadcastR   s   z#Equal.export_equal_string_broadcastN)r   r   )__name__
__module____qualname__staticmethodr2   r4   r=   r?   r0   r0   r0   r1   r      s    *r   )

__future__r   numpyr!   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r0   r0   r0   r1   <module>   s   