U
    rh                     @  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 )IsNaNNone)returnc                  C  s^   t jjddgdgd} tjdtjtjdtj tjgtjd}t|}t	| |g|gdd	 d S )
Nr   xyinputsoutputs333333ffffff@ZdtypeZ
test_isnanr   r   name)
onnxhelper	make_nodenparraynaninfZfloat32isnanr   noder   r	    r   E/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/isnan.pyexport   s    &
zIsNaN.exportc                  C  s^   t jjddgdgd} tjdtjtjdtj tjgtjd}t|}t	| |g|gdd	 d S )
Nr   r   r	   r
   r   r   r   Ztest_isnan_float16r   )
r   r   r   r   r   r   r   Zfloat16r   r   r   r   r   r   export_float16   s    &
zIsNaN.export_float16N)__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
   