
    /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                      e Zd Zedd            Zedd            Zedd            Zedd            Zedd            Zedd            Z	edd	            Z
edd
            ZdS )SlicereturnNonec                 *   t           j                            dg ddg          } t          j                            ddd                              t          j                  }|dd	ddf         }t          j        ddgt          j	        
          }t          j        d	dgt          j	        
          }t          j        ddgt          j	        
          }t          j        ddgt          j	        
          }t          | |||||g|gd           d S )Nr   xstartsendsaxesstepsyinputsoutputs   
      r      dtype   
test_slicer   r   nameonnxhelper	make_nodenprandomrandnastypefloat32arrayint64r   )noder   r   r   r   r   r   s          f/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/slice.pyexport_slicezSlice.export_slice   s   {$$;;;E % 
 
 IOOBA&&--bj99ac1R4iL1a&111xBrx000xAbh///!Qrx000!VT47!<	
 	
 	
 	
 	
 	
    c                 "   t           j                            dg ddg          } t          j                            ddd                              t          j                  }t          j        dgt          j	        	          }t          j        d
gt          j	        	          }t          j        dgt          j	        	          }t          j        dgt          j	        	          }|d d dd
f         }t          | |||||g|gd           d S )Nr   r   r   r   r   r   r   r   r   r   test_slice_negr   r   r*   r   r   r   r   r   r   s          r+   export_slice_negzSlice.export_slice_neg!   s   {$$;;;E % 
 
 IOOBA&&--bj991#RX...xBH---x28,,,!BH---aaa2gJvtT51C!		
 	
 	
 	
 	
 	
r-   c                 "   t           j                            dg ddg          } t          j                            ddd                              t          j                  }t          j        dgt          j	        	          }t          j        dgt          j	        	          }t          j        d
gt          j	        	          }t          j        d
gt          j	        	          }|d d ddf         }t          | |||||g|gd           d S )Nr   r   r   r   r   r   r     r   r   test_slice_start_out_of_boundsr   r   r1   s          r+    export_slice_start_out_of_boundsz&Slice.export_slice_start_out_of_bounds7   s   {$$;;;E % 
 
 IOOBA&&--bj994&111xbh///x28,,,!BH---aaadlOvtT51C1		
 	
 	
 	
 	
 	
r-   c                 "   t           j                            dg ddg          } t          j                            ddd                              t          j                  }t          j        dgt          j	        	          }t          j        d
gt          j	        	          }t          j        dgt          j	        	          }t          j        dgt          j	        	          }|d d dd
f         }t          | |||||g|gd           d S )Nr   r   r   r   r   r   r   r   r   r4   test_slice_end_out_of_boundsr   r   r1   s          r+   export_slice_end_out_of_boundsz$Slice.export_slice_end_out_of_boundsM   s   {$$;;;E % 
 
 IOOBA&&--bj991#RX...xbh///x28,,,!BH---aaa4iLvtT51C/		
 	
 	
 	
 	
 	
r-   c                    t           j                            dg ddg          } t          j                            ddd                              t          j                  }t          j        g dt          j	        	          }t          j        g d
t          j	        	          }|d d d d ddf         }t          | |||g|gd           d S )Nr   )r   r   r   r   r   r   r   r   r   r   r   r   r   r      r   r=   test_slice_default_axesr   r   )r*   r   r   r   r   s        r+   export_slice_default_axeszSlice.export_slice_default_axesc   s    {$$***E % 
 
 IOOBA&&--bj99)))28444x28444aaaAaCiL!VT*QC>W	
 	
 	
 	
 	
 	
r-   c                    t           j                            dg ddg          } t          j                            ddd                              t          j                  }t          j        g dt          j	        	          }t          j        g d
t          j	        	          }t          j        g dt          j	        	          }|d d d d ddf         }t          | ||||g|gd           d S )Nr   r   r   r   r   r   r   r   r   r   r;   r   r<   r   r      r   r=   test_slice_default_stepsr   r   r*   r   r   r   r   r   s         r+   export_slice_default_stepsz Slice.export_slice_default_stepst   s    {$$222E % 
 
 IOOBA&&--bj99)))28444x28444x			222aaaAaCiLvtT*C+		
 	
 	
 	
 	
 	
r-   c                 X   t           j                            dg ddg          } t          j                            ddd                              t          j                  }t          j        g dt          j	        	          }t          j        g d
t          j	        	          }t          j        g dt          j	        	          }t          j        g d                              t          j	                  }|dddddddddf         }t          | |||||g|gd           d S )Nr   r   r   r   r   r   r   r<   r   )r   r   r   rB   )r/   r   r/   rH   r=   r   rI   test_slice_neg_stepsr   r   r1   s          r+   export_slice_neg_stepszSlice.export_slice_neg_steps   s&   {$$;;;E % 
 
 IOOBA&&--bj99+++RX666x			222x			222&&--bh77b2gr!Bw!B&'vtT51C'		
 	
 	
 	
 	
 	
r-   c                    t           j                            dg ddg          } t          j                            ddd                              t          j                  }t          j        g dt          j	        	          }t          j        g d
t          j	        	          }t          j        g dt          j	        	          }|d d d d ddf         }t          | ||||g|gd           d S )Nr   rA   r   r   r   r   r   r;   r   r<   )r   rI   r/   r   r=   test_slice_negative_axesr   r   rE   s         r+   export_slice_negative_axesz Slice.export_slice_negative_axes   s    {$$222E % 
 
 IOOBA&&--bj99)))28444x28444x28444aaaAaCiLvtT*C+		
 	
 	
 	
 	
 	
r-   N)r   r	   )__name__
__module____qualname__staticmethodr,   r2   r6   r9   r?   rF   rK   rN    r-   r+   r   r      s       
 
 
 \
$ 
 
 
 \
* 
 
 
 \
* 
 
 
 \
* 
 
 
 \
  
 
 
 \
( 
 
 
 \
* 
 
 
 \
 
 
r-   r   )

__future__r   numpyr#   r    onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rS   r-   r+   <module>rX      s    # " " " " "      , , , , , , . . . . . .e
 e
 e
 e
 e
D e
 e
 e
 e
 e
r-   