
    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)Compressc                 n   t         j                  j                  dddgdgd      } t        j                  ddgd	d
gddgg      j                  t        j                        }t        j                  g d      }t        j                  ||d      }t        | ||j                  t              g|gd       y )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       a/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/compress.pyexport_compress_0zCompress.export_compress_0   s    {{$$[)J	 % 
 1a&1a&1a&1299"**EHHY'	YA6
 	9++D12H"		
    c                 n   t         j                  j                  dddgdgd      } t        j                  ddgdd	gd
dgg      j                  t        j                        }t        j                  ddg      }t        j                  ||d      }t        | ||j                  t              g|gd       y )Nr   r	   r
   r   r   r   r   r   r   r   r   r   r   test_compress_1r   r   r$   s       r&   export_compress_1zCompress.export_compress_1$   s    {{$$[)J	 % 
 1a&1a&1a&1299"**EHHaV$	YA6 	9++D12H"		
r(   c                 h   t         j                  j                  dddgdg      } t        j                  ddgdd	gd
dgg      j                  t        j                        }t        j                  g d      }t        j                  ||      }t        | ||j                  t              g|gd       y )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_axisz%Compress.export_compress_default_axis;   s    {{$$[)J % 

 1a&1a&1a&1299"**EHH_-	Y. 	9++D12H-		
r(   c                 n   t         j                  j                  dddgdgd      } t        j                  ddgd	d
gddgg      j                  t        j                        }t        j                  ddg      }t        j                  ||d      }t        | ||j                  t              g|gd       y )Nr   r	   r
   r   r   r   r   r   r   r   r   r   r   test_compress_negative_axisr   r   r$   s       r&   export_compress_negative_axisz&Compress.export_compress_negative_axisO   s    {{$$[)J	 % 
 1a&1a&1a&1299"**EHHaV$	YB7
 	9++D12H.		
r(   N)returnNone)__name__
__module____qualname__staticmethodr'   r+   r.   r2    r(   r&   r   r      sP    
 
* 
 
, 
 
& 
 
r(   r   )

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