
    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 )CompressreturnNc                  |   t           j                            dddgdgd          } t          j        ddgd	d
gddgg                              t          j                  }t          j        g d          }t          j        ||d          }t          | ||                    t                    g|gd           d S )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/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/compress.pyexport_compress_0zCompress.export_compress_0   s    {$$[)J	 % 
 
 1a&1a&1a&12299"*EEHYYY''	YA666
 	9++D112H"		
 	
 	
 	
 	
 	
    c                  |   t           j                            dddgdgd          } t          j        ddgdd	gd
dgg                              t          j                  }t          j        ddg          }t          j        ||d          }t          | ||                    t                    g|gd           d S )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299"*EEHaV$$	YA666 	9++D112H"		
 	
 	
 	
 	
 	
r(   c                  v   t           j                            dddgdg          } t          j        ddgdd	gd
dgg                              t          j                  }t          j        g d          }t          j        ||          }t          | ||                    t                    g|gd           d S )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299"*EEH___--	Y.. 	9++D112H-		
 	
 	
 	
 	
 	
r(   c                  |   t           j                            dddgdgd          } t          j        ddgd	d
gddgg                              t          j                  }t          j        ddg          }t          j        ||d          }t          | ||                    t                    g|gd           d S )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_axisN   s    {$$[)J	 % 
 
 1a&1a&1a&12299"*EEHaV$$	YB777
 	9++D112H.		
 	
 	
 	
 	
 	
r(   )r   N)__name__
__module____qualname__staticmethodr'   r+   r.   r2    r(   r&   r   r      s        
 
 
 \
* 
 
 
 \
, 
 
 
 \
& 
 
 
 \
 
 
r(   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r7   r(   r&   <module>r;      s~   
      , , , , , , . . . . . .V
 V
 V
 V
 V
t V
 V
 V
 V
 V
r(   