U
    r±ËhÈ	  ã                   @  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                   @  s0   e Zd Zeddœdd„ƒZeddœdd„ƒZdS )ÚAndÚNone)Úreturnc                  C  s  t jjdddgdgd} tj dd¡dk t¡}tj dd¡dk t¡}t ||¡}t	| ||g|gd	d
 tj ddd¡dk t¡}tj ddd¡dk t¡}t ||¡}t	| ||g|gdd
 tj dddd¡dk t¡}tj dddd¡dk t¡}t ||¡}t	| ||g|gdd
 d S )Nr   ÚxÚyÚand©ÚinputsÚoutputsé   é   r   Z
test_and2d©r   r   Únameé   Z
test_and3dé   Z
test_and4d©
ÚonnxÚhelperZ	make_nodeÚnpÚrandomZrandnZastypeÚboolÚlogical_andr   ©Únoder   r	   Úz© r   úC/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/and.pyÚexport   s"    ýz
And.exportc                  C  sÊ  t jjdddgdgd} tj ddd¡d	k t¡}tj d¡d	k t¡}t ||¡}t	| ||g|gd
d tj ddd¡d	k t¡}tj dd¡d	k t¡}t ||¡}t	| ||g|gdd tj dddd¡d	k t¡}tj dd¡d	k t¡}t ||¡}t	| ||g|gdd tj dddd¡d	k t¡}tj ddd¡d	k t¡}t ||¡}t	| ||g|gdd tj dddd¡d	k t¡}tj dddd¡d	k t¡}t ||¡}t	| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   Ztest_and_bcast3v1dr   Ztest_and_bcast3v2dr   Ztest_and_bcast4v2dZtest_and_bcast4v3dé   Ztest_and_bcast4v4dr   r   r   r   r   Úexport_and_broadcast(   s2    ýzAnd.export_and_broadcastN)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr    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
   