
    bIg                    N    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	y)    )annotationsN)Base)expectc                  0    e Zd Zedd       Zedd       Zy)Expandc                    t         j                  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
       y )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/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/expand.pyexport_dim_changedzExpand.export_dim_changed   s    {{$$K(L % 

 Azz"))Arwwu~'9LeT 	"'')2::>> HHYbhh7	)$J*		
    c                    t         j                  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
       y )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	zz"))Arwwu~'9LeT 774#
 HHYbhh7	)$J,		
r'   N)returnNone)__name__
__module____qualname__staticmethodr&   r,    r'   r%   r   r      s(    
 
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
T 5
r'   