
    /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dS )ExpandreturnNonec                    t           j                            dddgdg          } ddg}t          j        t          j        dt          j        |          dz   t          j                  |          }g d	}|t          j        |t          j                  z  }t          j	        |t          j
                  }t          | ||g|gd
           d S )Nr   data	new_shapeexpandedinputsoutputs      dtype)   r      test_expand_dim_changedr   r   name)onnxhelper	make_nodenpreshapearangeprodfloat32onesarrayint64r   )nodeshaper   r   r   s        g/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/expand.pyexport_dim_changedzExpand.export_dim_changed   s    {$$K(L % 
 

 Az")Arwu~~'9LLLeTT II	"')2:>>>> HYbh777	)$J*		
 	
 	
 	
 	
 	
    c                    t           j                            dddgdg          } ddg}ddg}t          j        t          j        dt          j        |          dz   t          j        	          |          }t          j        |d          }t          j	        |t          j
        	          }t          | ||g|gd
           d S )Nr   r   r   r   r   r   r      r   test_expand_dim_unchangedr   )r   r   r   r   r   r   r    r!   tiler#   r$   r   )r%   r&   r   r   r   s        r'   export_dim_unchangedzExpand.export_dim_unchanged+   s    {$$K(L % 
 

 AF	z")Arwu~~'9LLLeTT 74##
 HYbh777	)$J,		
 	
 	
 	
 	
 	
r)   N)r   r	   )__name__
__module____qualname__staticmethodr(   r.    r)   r'   r   r      sR        
 
 
 \
8 
 
 
 \
 
 
r)   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r3   r)   r'   <module>r8      s    # " " " " "      , , , , , , . . . . . .5
 5
 5
 5
 5
T 5
 5
 5
 5
 5
r)   