
    Ngs                     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                       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returnNc                  *   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          ]/var/www/html/ai-engine/env/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   r/   s          r)    export_slice_start_out_of_boundsz&Slice.export_slice_start_out_of_bounds6   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   r2   test_slice_end_out_of_boundsr   r   r/   s          r)   export_slice_end_out_of_boundsz$Slice.export_slice_end_out_of_boundsL   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_axesb   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   r9   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_stepss   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   r@   )r-   r   r-   rF   r;   r   rG   test_slice_neg_stepsr   r   r/   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   r?   r   r   r   r   r   r9   r   r:   )r   rG   r-   r   r;   test_slice_negative_axesr   r   rC   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+   )r   N)__name__
__module____qualname__staticmethodr*   r0   r4   r7   r=   rD   rI   rL    r+   r)   r   r      s       
 
 
 \
$ 
 
 
 \
* 
 
 
 \
* 
 
 
 \
* 
 
 
 \
  
 
 
 \
( 
 
 
 \
* 
 
 
 \
 
 
r+   r   )numpyr!   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rQ   r+   r)   <module>rU      s~   
      , , , , , , . . . . . .e
 e
 e
 e
 e
D e
 e
 e
 e
 e
r+   