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 )
EyeLikeNone)returnc                  C  sb   d} t jjddgdgd}tjjdd| tjd}tj| d | d	 tjd
}t||g|gdd d S )N)   r   r   xy)inputsoutputsr   d   sizedtype   r   Ztest_eyelike_without_dtyper   r   name)	onnxhelper	make_nodenprandomrandintint32eyer   shapenoder	   r
    r    G/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/eyelike.pyexport_without_dtype   s    zEyeLike.export_without_dtypec                  C  sh   d} t jjddgdgt jjd}tjjdd| tjd}tj	| d | d	 tj
d
}t||g|gdd d S )N)   r   r   r	   r
   )r   r   r   r   r   r   r   r   Ztest_eyelike_with_dtyper   )r   r   r   TensorProtoZDOUBLEr   r   r   r   r   Zfloat64r   r   r    r    r!   export_with_dtype   s    zEyeLike.export_with_dtypec                  C  sp   d} d}t jjddgdg|t jjd}tjjdd| tjd	}tj	| d | d |tj
d
}t||g|gdd d S )N)r      r   r   r	   r
   )r   r   kr   r   r   r   )r'   r   Z'test_eyelike_populate_off_main_diagonalr   )r   r   r   r$   FLOATr   r   r   r   r   Zfloat32r   )r   Zoff_diagonal_offsetr   r	   r
   r    r    r!   !export_populate_off_main_diagonal)   s"    z)EyeLike.export_populate_off_main_diagonalN)__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
   