
    bIg                    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                  T    e Zd Zedd       Zedd       Zedd       Zedd       Zy)IsInfc                 `   t         j                  j                  ddgdg      } t        j                  dt        j
                  t        j                  dt        j                   t        j                  gt        j                        }t        j                  |      }t        | |g|gd	       y )
Nr   xyinputsoutputs333333ffffff@dtype
test_isinfr   r   name)
onnxhelper	make_nodenparraynaninffloat32isinfr   noder	   r
   s      ^/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/isinf.pyexport_infinityzIsInf.export_infinity   s|    {{$$5E % 
 HHdBFFBFFC"&&"&&ATHHQKtQC!<@    c                 b   t         j                  j                  ddgdgd      } t        j                  dt        j
                  t        j                  dt        j                   t        j                  gt        j                        }t        j                  |      }t        | |g|gd	
       y )Nr   r	   r
   r   )r   r   detect_negative333333g@r   test_isinf_positiver   )
r   r   r   r   r   r   r   r   isposinfr   r   s      r    export_positive_infinity_onlyz#IsInf.export_positive_infinity_only   s}    {{$$SEC5! % 
 HHdBFFBFFC"&&"&&ATKKNtQC!3HIr"   c                 b   t         j                  j                  ddgdgd      } t        j                  dt        j
                  t        j                  dt        j                   t        j                  gt        j                        }t        j                  |      }t        | |g|gd	
       y )Nr   r	   r
   r   )r   r   detect_positiver%   gr   test_isinf_negativer   )
r   r   r   r   r   r   r   r   isneginfr   r   s      r    export_negative_infinity_onlyz#IsInf.export_negative_infinity_only$   s}    {{$$SEC5! % 
 HHdBFFBFFD266'266B"**UKKNtQC!3HIr"   c                 `   t         j                  j                  ddgdg      } t        j                  dt        j
                  t        j                  dt        j                   t        j                  gt        j                        }t        j                  |      }t        | |g|gd	       y )
Nr   r	   r
   r   r   r   r   test_isinf_float16r   )
r   r   r   r   r   r   r   float16r   r   r   s      r    export_infinity_float16zIsInf.export_infinity_float16.   s}    {{$$5E % 
 HHdBFFBFFC"&&"&&ATHHQKtQC!3GHr"   N)returnNone)__name__
__module____qualname__staticmethodr!   r(   r-   r1    r"   r    r   r      sX    	A 	A J J J J 	I 	Ir"   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r8   r"   r    <module>r=      s$    #   , .+ID +Ir"   