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                   @  sB   e Zd ZeddddZeddddZeddddZd	S )
AddNone)returnc                  C  sf   t jjdddgdgd} tjdddtj}tjdddtj}t| ||g|| gd	d
 d S )Nr   xysuminputsoutputs         Ztest_addr   r   name	onnxhelper	make_nodenprandomZrandnZastypeZfloat32r   noder   r	    r   C/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/add.pyexport   s    z
Add.exportc                  C  s^   t jjdddgdgd} tjjddtjd}tjjddtjd}t| ||g|| gd	d
 d S )Nr   r   r	   r
   r      )r   r   r   )sizeZdtypeZtest_add_uint8r   )r   r   r   r   r   randintZuint8r   r   r   r   r   export_add_uint8   s    zAdd.export_add_uint8c                  C  sb   t jjdddgdgd} tjdddtj}tjdtj}t| ||g|| gd	d
 d S )Nr   r   r	   r
   r   r   r   r   Ztest_add_bcastr   r   r   r   r   r   export_add_broadcast&   s    zAdd.export_add_broadcastN)__name__
__module____qualname__staticmethodr   r!   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
   