
    bIg2                    N    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e      Z	y)    )annotationsN)Base)expectc                  0    e Zd Zedd       Zedd       Zy)Wherec                    t         j                  j                  dg ddg      } t        j                  ddgddggt
              }t        j                  ddgd	d
ggt        j                        }t        j                  ddgddggt        j                        }t        j                  |||      }t        | |||g|gd       y )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   s        ^/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/where.pyexportzWhere.export   s    {{$$*E % 
 HHq!fq!f-T:	HHq!fq!f%RZZ8HHq!fq!f%RZZ8HHY1%tY1-sAUV    c                    t         j                  j                  dg ddg      } t        j                  ddgddggt
              }t        j                  ddgd	d
ggt        j                        }t        j                  ddgddggt        j                        }t        j                  |||      }t        | |||g|gd       y )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&   s        r(   export_longzWhere.export_long   s    {{$$*E % 
 HHq!fq!f-T:	HHq!fq!f%RXX6HHq!fq!f%RXX6HHY1%)Q*QC>W	
r*   N)returnNone)__name__
__module____qualname__staticmethodr)   r.    r*   r(   r   r      s*    W W 
 
r*   r   )

__future__r   numpyr!   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r5   r*   r(   <module>r:      s"    #   , .
D 
r*   