o
    q::j                     @  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 )	DivreturnNonec                  C  s  t jjdddgdgd} tddgtj}tdd	gtj}|| }t| ||g|gd
d tj	dddtj}tj
dddtjd }|| }t| ||g|gdd tjjddtjd}tjjddtjdd }|| }t| ||g|gdd tjjddtjd}tjjddtjdd }|| }t| ||g|gdd tjg dtjd}tjg dtjd}tjg dtjd}t| ||g|gdd tjjddtjd}tjjddtjdd }|| }t| ||g|gdd tjjddtjd}tjjddtjdd }|| }t| ||g|gdd tjjddtjd}tjjddtjdd }|| }t| ||g|gdd tjjddtjd}tjjddtjdd }|| }t| ||g|gdd d S )Nr   xyzinputsoutputs            test_div_exampler   r   name         ?test_div   )r   r   r   )sizedtypetest_div_int8test_div_int16)r   r   r   )r   )r   r   r   )r   r   r   test_div_int32_trunctest_div_uint8test_div_uint16test_div_uint32test_div_uint64)onnxhelper	make_nodenparrayastypefloat32r   randomrandnrandrandintint8int16int32uint8uint16uint32uint64noder   r	   r
    r9   f/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/div.pyexport   sR   z
Div.exportc                  C  sj   t jjdddgdgd} tjdddtj}tjdtjd	 }|| }t	| ||g|gd
d d S )Nr   r   r	   r
   r   r   r   r   r   test_div_bcastr   )
r%   r&   r'   r(   r,   r-   r*   r+   r.   r   r7   r9   r9   r:   export_div_broadcastC   s   zDiv.export_div_broadcastN)r   r   )__name__
__module____qualname__staticmethodr;   r=   r9   r9   r9   r:   r      s
    4r   )

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