o
    soi2                     @  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                   @  s(   e Zd ZedddZedddZdS )	WherereturnNonec                  C     t jjdg d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   dtype         	            test_where_exampler   r   name)	onnxhelper	make_nodenparrayboolfloat32wherer   noder
   r   r   r    r(   U/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/where.pyexport   s   zWhere.exportc                  C  r   )Nr   r	   r   r   r   r   r   r   r   r   r   r   r   r   test_where_long_exampler   )	r   r   r    r!   r"   r#   int64r%   r   r&   r(   r(   r)   export_long   s   
zWhere.export_longN)r   r   )__name__
__module____qualname__staticmethodr*   r-   r(   r(   r(   r)   r      s
    r   )

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