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 )RangeNone)returnc                  C  sh   t jjddddgdgd} td}td}td	}tj|||tjd
}t| |||g|gdd d S )Nr   startlimitdeltaoutputinputsoutputs         ZdtypeZ$test_range_float_type_positive_deltar   r   name)onnxhelper	make_nodenpZfloat32aranger   noder   r	   r
   r    r   G/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/rangeop.py&export_range_float_type_positive_delta   s(    


   z,Range.export_range_float_type_positive_deltac                  C  sh   t jjddddgdgd} td}td}td	}tj|||tjd
}t| |||g|gdd d S )Nr   r   r	   r
   r   r   
      r   Z$test_range_int32_type_negative_deltar   )r   r   r   r   Zint32r   r   r   r   r   r   &export_range_int32_type_negative_delta$   s(    


   z,Range.export_range_int32_type_negative_deltaN)__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
   