o
    to™i  ã                	   @  sb   d dl mZ d dlZd dlZd dlmZ ej ej ej 	ej e
¡¡¡d¡Zedfddd„ZdS )é    )ÚannotationsN)ÚTestCaseÚdataÚdata_dirÚstrÚkindú
str | NoneÚreturnúlist[TestCase]c                 C  sˆ  t  | ¡}||vrtd|› ƒ‚g }t j | |¡}t  |¡D ]£}t j ||¡}d}d}t j |¡s2qt j t j |d¡¡r}d}	|tdƒ }
|}t j t j |d¡¡r|tt j |d¡ƒ}t	 
|¡}| d|¡}| d	|¡}W d  ƒ n1 sww   Y  n4tt j |d¡ƒ#}t	 
|¡}|d
 }	|d }
| d|¡}| d	|¡}d}W d  ƒ n1 s¬w   Y  | t||	|
|dd|||d	¡ q|S )z.Load model test cases from on-disk data files.zkind must be one of gü©ñÒMbP?gH¯¼šò×z>z
model.onnxNÚtest_z	data.jsonÚrtolÚatolÚurlÚ
model_name)	Únamer   r   Ú	model_dirÚmodelÚ	data_setsr   r   r   )ÚosÚlistdirÚ
ValueErrorÚpathÚjoinÚisdirÚexistsÚlenÚopenÚjsonÚloadÚgetÚappendr   )r   r   Úsupported_kindsÚ	testcasesÚkind_dirÚ	test_nameÚcase_dirr   r   r   r   r   Úfr   © r'   úU/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/loader/__init__.pyÚload_model_tests   sX   

ý€
ú÷ÿr)   )r   r   r   r   r	   r
   )Ú
__future__r   r   r   Ú onnx.backend.test.case.test_caser   r   r   ÚdirnameÚrealpathÚ__file__ÚDATA_DIRr)   r'   r'   r'   r(   Ú<module>   s   ÿþ