
    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)	Compress   Nc                  ^   [         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       ^/var/www/fran/franai/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_axisN   s    {{$$[)J	 % 
 1a&1a&1a&1299"**EHHaV$	YB7
 	++D12H.		
r)    )returnN)
__name__
__module____qualname____firstlineno__staticmethodr'   r,   r0   r5   __static_attributes__r7   r)   r&   r   r      sP    
 
* 
 
, 
 
& 
 
r)   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r7   r)   r&   <module>rB      s!   
   , .V
t V
r)   