
    hx                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   <    \ rS rSr\SS j5       r\SS j5       rSrg)Expand   Nc                     [         R                  R                  SSS/S/S9n SS/n[        R                  " [        R
                  " S[        R                  " U5      S-   [        R                  S9U5      n/ S	QnU[        R                  " U[        R                  S9-  n[        R                  " U[        R                  S9n[        U X#/U/S
S9  g )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        \/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/expand.pyexport_dim_changedExpand.export_dim_changed   s    {{$$K(L % 

 Azz"))Arwwu~'9LeT 	"'')2::>> HHYbhh7	$J*		
    c                     [         R                  R                  SSS/S/S9n SS/nSS/n[        R                  " [        R
                  " S[        R                  " U5      S-   [        R                  S	9U5      n[        R                  " US5      n[        R                  " U[        R                  S	9n[        U X2/U/S
S9  g )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_unchangedExpand.export_dim_unchanged*   s    {{$$K(L % 

 AF	zz"))Arwwu~'9LeT 774#
 HHYbhh7	$J,		
r(    )returnN)__name__
__module____qualname____firstlineno__staticmethodr&   r-   __static_attributes__r/   r(   r%   r   r      s(    
 
8 
 
r(   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r/   r(   r%   <module>r:      s   
   , .5
T 5
r(   