U
    rh                     @  sH   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 deZ	dS )    )annotationsN)Base)expectc                   @  s0   e Zd ZeddddZeddddZdS )ExpandNone)returnc                  C  s   t jjdddgdgd} ddg}ttjdt|d tjd|}d	dd
g}|tj|tjd }tj	|tj
d}t| ||g|gdd d S )Nr   data	new_shapeexpandedinputsoutputs      Zdtype      Ztest_expand_dim_changedr   r   name)onnxhelper	make_nodenpreshapearangeprodfloat32Zonesarrayint64r   )nodeshaper   r	   r
    r!   F/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/expand.pyexport_dim_changed   s     $
	zExpand.export_dim_changedc                  C  s   t jjdddgdgd} ddg}ddg}ttjdt|d tjd	|}t|d}tj	|tj
d	}t| ||g|gd
d d S )Nr   r   r	   r
   r   r   r      r   Ztest_expand_dim_unchangedr   )r   r   r   r   r   r   r   r   Ztiler   r   r   )r   r    r	   r   r
   r!   r!   r"   export_dim_unchanged+   s     $zExpand.export_dim_unchangedN)__name__
__module____qualname__staticmethodr#   r%   r!   r!   r!   r"   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r!   r!   r!   r"   <module>   s
   