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 )SqueezeNone)returnc                  C  sl   t jjdddgdgd} tjdddd	tj}tjd
gtj	d}tj
|d
d}t| ||g|gdd d S )Nr   xaxesyinputsoutputs            r   ZdtypeZaxisZtest_squeezer   r   nameonnxhelperZ	make_nodenprandomZrandnZastypeZfloat32arrayZint64Zsqueezer   noder   r	   r
    r   G/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/squeeze.pyexport_squeeze   s    zSqueeze.export_squeezec                  C  sl   t jjdddgdgd} tjddddtj}tjd	gtj	d
}tj
|d	d}t| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   r   Ztest_squeeze_negative_axesr   r   r   r   r   r   export_squeeze_negative_axes   s    z$Squeeze.export_squeeze_negative_axesN)__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
   