o
    soi                     @  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                   @  s(   e Zd ZedddZedddZdS )	ExpandreturnNonec                  C  s   t jjdddgdgd} ddg}ttjdt|d tjd|}g d	}|tj|tjd }tj	|tj
d}t| ||g|gd
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
    r$   V/home/ubuntu/.local/lib/python3.10/site-packages/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   test_expand_dim_unchangedr   )r   r   r   r   r   r   r   r   tiler    r!   r   )r"   r#   r	   r   r
   r$   r$   r%   export_dim_unchanged+   s    $
zExpand.export_dim_unchangedN)r   r   )__name__
__module____qualname__staticmethodr&   r*   r$   r$   r$   r%   r      s
    r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r$   r$   r$   r%   <module>   s   