U
    rh2                     @  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 )WhereNone)returnc                  C  s   t jjddddgdgd} tjddgddggtd	}tjdd
gddggtjd	}tjddgddggtjd	}t|||}t| |||g|gdd d S )Nr   	conditionxyzinputsoutputs   r   Zdtype         	            Ztest_where_exampler   r   name)	onnxhelper	make_nodenparrayboolZfloat32wherer   noder   r	   r
   r    r#   E/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/where.pyexport   s    zWhere.exportc                  C  s   t jjddddgdgd} tjddgddggtd	}tjdd
gddggtjd	}tjddgddggtjd	}t|||}t| |||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   Ztest_where_long_exampler   )	r   r   r   r   r   r   Zint64r    r   r!   r#   r#   r$   export_long   s       zWhere.export_longN)__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
   