
    h                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   d    \ rS rSr\SS j5       r\SS j5       r\SS j5       r\SS j5       rSr	g)	IsInf   Nc                  N   [         R                  R                  SS/S/S9n [        R                  " S[        R
                  [        R                  S[        R                  [        R                  /[        R                  S9n[        R                  " U5      n[        X/U/SS	9  g )
Nr   xyinputsoutputs333333ffffff@dtype
test_isinfr   r   name)onnxhelper	make_nodenparraynaninfNINFfloat32isinfr   noder	   r
   s      [/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/isinf.pyexport_infinityIsInf.export_infinity   sx    {{$$5E % 
 HHdBFFBFFC"&&ATHHQKtC!<@    c                  P   [         R                  R                  SS/S/SS9n [        R                  " S[        R
                  [        R                  S[        R                  [        R                  /[        R                  S9n[        R                  " U5      n[        X/U/S	S
9  g )Nr   r	   r
   r   )r   r   detect_negative333333g@r   test_isinf_positiver   )r   r   r   r   r   r   r   r   r   isposinfr   r   s      r!   export_positive_infinity_only#IsInf.export_positive_infinity_only   sy    {{$$SEC5! % 
 HHdBFFBFFC"&&ATKKNtC!3HIr$   c                  P   [         R                  R                  SS/S/SS9n [        R                  " S[        R
                  [        R                  S[        R                  [        R                  /[        R                  S9n[        R                  " U5      n[        X/U/S	S
9  g )Nr   r	   r
   r   )r   r   detect_positiver'   gr   test_isinf_negativer   )r   r   r   r   r   r   r   r   r   isneginfr   r   s      r!   export_negative_infinity_only#IsInf.export_negative_infinity_only#   sy    {{$$SEC5! % 
 HHdBFFBFFD"''266B"**UKKNtC!3HIr$   c                  N   [         R                  R                  SS/S/S9n [        R                  " S[        R
                  [        R                  S[        R                  [        R                  /[        R                  S9n[        R                  " U5      n[        X/U/SS	9  g )
Nr   r	   r
   r   r   r   r   test_isinf_float16r   )r   r   r   r   r   r   r   r   float16r   r   r   s      r!   export_infinity_float16IsInf.export_infinity_float16-   sy    {{$$5E % 
 HHdBFFBFFC"&&ATHHQKtC!3GHr$    )returnN)
__name__
__module____qualname____firstlineno__staticmethodr"   r*   r0   r5   __static_attributes__r7   r$   r!   r   r      sX    	A 	A J J J J 	I 	Ir$   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r7   r$   r!   <module>rB      s!   
   , .+ID +Ir$   