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 )GreaterNone)returnc                  C  sn   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||}t	| ||g|gd	d
 d S )NGreaterOrEqualxygreater_equalinputsoutputs         Ztest_greater_equalr   r   name
onnxhelperZ	make_nodenprandomZrandnZastypeZfloat32r   r   noder	   r
   z r   M/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/greater_equal.pyexport   s    zGreater.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}t||}t	| ||g|gd	d
 d S )Nr   r	   r
   r   r   r   r   r   Ztest_greater_equal_bcastr   r   r   r   r   r   export_greater_broadcast   s    z Greater.export_greater_broadcastN)__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
   