o
    wi                     @  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                   @  sD   e Zd ZedddZedddZedddZedd	d
ZdS )IsInfreturnNonec                  C  ^   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@dtype
test_isinfr   r   name)
onnxhelper	make_nodenparraynaninffloat32isinfr   noder	   r
    r    ^/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/isinf.pyexport_infinity      &
zIsInf.export_infinityc                  C  `   t jjddgdgd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   detect_negative333333g@r   test_isinf_positiver   )
r   r   r   r   r   r   r   r   isposinfr   r   r    r    r!   export_positive_infinity_only      &
z#IsInf.export_positive_infinity_onlyc                  C  r$   )Nr   r	   r
   r   )r   r   detect_positiver&   gr   test_isinf_negativer   )
r   r   r   r   r   r   r   r   isneginfr   r   r    r    r!   export_negative_infinity_only$   r*   z#IsInf.export_negative_infinity_onlyc                  C  r   )
Nr   r	   r
   r   r   r   r   test_isinf_float16r   )
r   r   r   r   r   r   r   float16r   r   r   r    r    r!   export_infinity_float16.   r#   zIsInf.export_infinity_float16N)r   r   )__name__
__module____qualname__staticmethodr"   r)   r.   r1   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   