o
    q::j                     @  sP   d dl mZ d dl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                   @  sD   e Zd ZedddZedddZedddZedd	d
ZdS )RangereturnNonec                  C  f   t jjdg d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         dtype$test_range_float_type_positive_deltar   r   name)onnxhelper	make_nodenpfloat32aranger   noder
   r   r   r    r!   j/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/rangeop.py&export_range_float_type_positive_delta   "   




z,Range.export_range_float_type_positive_deltac                  C  r   )Nr   r	   r   r   r   r   r   r   &test_range_float16_type_positive_deltar   )r   r   r   r   float16r   r   r   r!   r!   r"   (export_range_float16_type_positive_delta%   r$   z.Range.export_range_float16_type_positive_deltac                  C  s   t jjdg ddgd} tjdtjd}tjdtjd}tjdtjd}tjdddtjd	tj}t
| |||g|gd	d
 d S )Nr   r	   r   r   g      ?r   g      @g       @'test_range_bfloat16_type_positive_deltar   )r   r   r   r   array	ml_dtypesbfloat16r   r   astyper   r   r!   r!   r"   )export_range_bfloat16_type_positive_delta;   s"   
z/Range.export_range_bfloat16_type_positive_deltac                  C  r   )Nr   r	   r   r   
      r   $test_range_int32_type_negative_deltar   )r   r   r   r   int32r   r   r   r!   r!   r"   &export_range_int32_type_negative_deltaR   r$   z,Range.export_range_int32_type_negative_deltaN)r   r   )__name__
__module____qualname__staticmethodr#   r'   r-   r3   r!   r!   r!   r"   r      s    r   )
__future__r   r*   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r!   r!   r!   r"   <module>   s   