U
    qhY                     @  s:   d dl mZ d dlZG dd dejZedkr6e  dS )    )annotationsNc                   @  s   e Zd ZddddZdS )TestONNXRuntimeNone)returnc                 C  s  zdd l }~W n  tk
r.   tdd Y nX ddlm}m} ddl m} ddlm	} ddl
m}m}m}m}	 |d}
||
}|| |j|dd	 |j|dddd
}|	|d}|| dgd}| d j}| d j}|d}||}||g||i d S )Nr   zonnxruntime not installed)float32random)InferenceSession)get_example)checkerloadshape_inferenceversion_converterzsigmoid.onnxT)Z
full_check)Z
check_typeZstrict_modeZ	data_prop
   ZCPUExecutionProvider)Z	providers)         )onnxruntimeImportErrorunittestZSkipTestZnumpyr   r   r   Zonnxruntime.datasetsr	   Zonnxr
   r   r   r   Zcheck_modelZinfer_shapesZconvert_versionZSerializeToString
get_inputsnameget_outputsZastyperun)selfr   r   r   r   r	   r
   r   r   r   Zexample1modelZinferred_modelZconverted_modelsessZ
input_nameZoutput_namex r   ;/tmp/pip-unpacked-wheel-xnis5xre/onnx/test/test_with_ort.pytest_with_ort_example   s:    
    

z%TestONNXRuntime.test_with_ort_exampleN)__name__
__module____qualname__r   r   r   r   r   r      s   r   __main__)
__future__r   r   ZTestCaser   r    mainr   r   r   r   <module>   s   '