o
    wi?                     @  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 )	SqueezereturnNonec                  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   dtypeaxistest_squeezer   r   nameonnxhelper	make_nodenprandomrandnastypefloat32arrayint64squeezer   noder   r	   r
    r'   `/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/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   test_squeeze_negative_axesr   r   r%   r'   r'   r(   export_squeeze_negative_axes   s   z$Squeeze.export_squeeze_negative_axesN)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   