
    Si
                    N    S SK Jr  S SKrS SKrS SKJr  S SKJr   " S S\5      r	g)    )annotationsN)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)	Compress   c                 ^   [         R                  R                  SSS/S/SS9n [        R                  " SS/S	S
/SS//5      R                  [        R                  5      n[        R                  " / SQ5      n[        R                  " X!SS9n[        U XR                  [        5      /U/SS9  g )Nr   input	conditionoutputr   inputsoutputsaxis                  )r   r   r   r   test_compress_0r   r   name
onnxhelper	make_nodenparrayastypefloat32compressr   boolnoder
   r   r   s       c/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/compress.pyexport_compress_0Compress.export_compress_0   s    {{$$[)J	 % 
 1a&1a&1a&1299"**EHHY'	YA6
 	++D12H"		
    c                 ^   [         R                  R                  SSS/S/SS9n [        R                  " SS/SS	/S
S//5      R                  [        R                  5      n[        R                  " SS/5      n[        R                  " X!SS9n[        U XR                  [        5      /U/SS9  g )Nr   r
   r   r   r   r   r   r   r   r   r   r   r   test_compress_1r   r   r%   s       r'   export_compress_1Compress.export_compress_1$   s    {{$$[)J	 % 
 1a&1a&1a&1299"**EHHaV$	YA6 	++D12H"		
r*   c                 ^   [         R                  R                  SSS/S/S9n [        R                  " SS/SS	/S
S//5      R                  [        R                  5      n[        R                  " / SQ5      n[        R                  " X!5      n[        U XR                  [        5      /U/SS9  g )Nr   r
   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   test_compress_default_axisr   r   r%   s       r'   export_compress_default_axis%Compress.export_compress_default_axis;   s    {{$$[)J % 

 1a&1a&1a&1299"**EHH_-	Y. 	++D12H-		
r*   c                 ^   [         R                  R                  SSS/S/SS9n [        R                  " SS/S	S
/SS//5      R                  [        R                  5      n[        R                  " SS/5      n[        R                  " X!SS9n[        U XR                  [        5      /U/SS9  g )Nr   r
   r   r   r   r   r   r   r   r   r   r   r   test_compress_negative_axisr   r   r%   s       r'   export_compress_negative_axis&Compress.export_compress_negative_axisO   s    {{$$[)J	 % 
 1a&1a&1a&1299"**EHHaV$	YB7
 	++D12H.		
r*    N)returnNone)
__name__
__module____qualname____firstlineno__staticmethodr(   r-   r1   r6   __static_attributes__r8   r*   r'   r   r      sP    
 
* 
 
, 
 
& 
 
r*   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r8   r*   r'   <module>rE      s$    #   , .V
t V
r*   