
    /j
                    V    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	dS )    )annotationsN)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returnNonec                 |   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       i/home/longshao/multi-rider-rag/.venv/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_axisO   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*   N)r   r	   )__name__
__module____qualname__staticmethodr)   r-   r0   r4    r*   r(   r   r      s        
 
 
 \
* 
 
 
 \
, 
 
 
 \
& 
 
 
 \
 
 
r*   r   )

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