
    Ng                     J    d dl Zd dlZd dlmZ d dlmZ  G d de          ZdS )    N)Base)expectc                   n    e Zd Zedd            Zedd            Zedd            Zedd            ZdS )IsInfreturnNc                  >   t           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	           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
   s      ]/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/isinf.pyexport_infinityzIsInf.export_infinity   s    {$$5E % 
 
 HdBFBFC"&"&ATTTHQKKtQC!<@@@@@@    c                  @   t           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	
           d S )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! % 
 
 HdBFBFC"&"&ATTTKNNtQC!3HIIIIIIr"   c                  @   t           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	
           d S )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! % 
 
 HdBFBFD26'26B"*UUUKNNtQC!3HIIIIIIr"   c                  >   t           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	           d S )
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 % 
 
 HdBFBFC"&"&ATTTHQKKtQC!3GHHHHHHr"   )r   N)__name__
__module____qualname__staticmethodr!   r(   r-   r1    r"   r    r   r      s        	A 	A 	A \	A J J J \J J J J \J 	I 	I 	I \	I 	I 	Ir"   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r6   r"   r    <module>r:      s~   
      , , , , , , . . . . . .+I +I +I +I +ID +I +I +I +I +Ir"   