
    
Ng                        d dl Z d dlZd dlZd dlmZ d dlZd dlmZmZ ej	        
                    dej        vpej        d          d           G d dej                              Zed	k    r ej                     dS dS )
    N)join)
ModelProtohubTEST_HUBzConserving Git LFS quota)reasonc                   n    e Zd ZddZddZddZddZddZddZdd	Z	dd
Z
ddZddZddZddZdS )TestModelHubreturnNc                 0    d| _         d| _        d| _        d S )NMNISTzonnx/models:main   )namerepoopsetselfs    N/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/test/hub_test.pysetUpzTestModelHub.setUp   s    	&	


    c           	      J   t          j        | j        | j        d          }|                     |t
                     t          t          j        t          t          j	                    dd          d                    }| 
                    t          |          d           d S )NT)force_reload***.onnx	recursive   r   loadr   r   assertIsInstancer   listglobr   get_dirassertGreaterEquallenr   modelcached_filess      r   test_force_reloadzTestModelHub.test_force_reload   s    DIDAAAeZ000Id3;==$99TJJJ
 
 	L 1 1155555r   c                    t          j        | j        ddg          }t          j        | j        dg          }t          j        | j                  }|                     t	          |          d           |                     t	          |          t	          |                     |                     t	          |          t	          |                     d S )Nmnistvision)r&   tags)r,   r   )r   list_modelsr   assertGreaterr$   )r   model_info_list_1model_info_list_2model_info_list_3s       r   test_listing_modelsz TestModelHub.test_listing_models#   s    ODIWH:VVVODIXJGGGODI6630111555301137H3I3IJJJ301137H3I3IJJJJJr   c           	      F   t          j        | j        | j                  }|                     |t
                     t          t          j        t          t          j	                    dd          d                    }| 
                    t          |          d           d S )Nr   r   Tr   r   r   r%   s      r   test_basic_usagezTestModelHub.test_basic_usage,   s    DI..eZ000Id3;==$99TJJJ
 
 	L 1 1155555r   c           	         t          j                    }t          |d          }t          j        |           t          j        | j        | j                  }|                     |t                     t          t          j
        t          |dd          d                    }|                     t          |          d           t          j        |           d S )Ncustomr   r   Tr   r   )r   r"   r   set_dirr   r   r   r   r   r    r!   r#   r$   )r   	old_cache	new_cacher&   r'   s        r   test_custom_cachezTestModelHub.test_custom_cache5   s    KMM	H--	IDI..eZ000DId9dH&E&EQUVVVWWL 1 11555Ir   c                 ~    t          j        | j        | j        d          }|                     |t
                     d S )N   r   )r   r   r   r   r   r   r   r&   s     r   test_download_with_opsetz%TestModelHub.test_download_with_opsetB   s8    DIQ777eZ00000r   c                 D                           t           fd           d S )Nc                  F    t          j         j         j        d          S )Nr=   )r   r   r   r   r   s   r   <lambda>z/TestModelHub.test_opset_error.<locals>.<lambda>H   s    CHTY	$L$L$L r   assertRaisesAssertionErrorr   s   `r   test_opset_errorzTestModelHub.test_opset_errorF   s4    LLLL	
 	
 	
 	
 	
r   c                 D                           t           fd           d S )Nc                  <    t          j         j        dd          S )Nonnx/models:unknownT)silent)r   r   r   r   s   r   rC   z6TestModelHub.test_manifest_not_found.<locals>.<lambda>N   s    CHTY(=dKKK r   rD   r   s   `r   test_manifest_not_foundz$TestModelHub.test_manifest_not_foundK   s4    KKKK	
 	
 	
 	
 	
r   c                    t          j        d          }|                     |           t          j        d          }|                     |           t          j        | j                  }|                     |           d S )Nzmhamilton723/modelsrJ   )r   _verify_repo_refassertFalser   
assertTrue)r   verifieds     r   test_verify_repo_refz!TestModelHub.test_verify_repo_refQ   sz    '(=>>""" '(=>>""" '	22!!!!!r   c                      t          j        d j        d           t          j        d j                                        t           fd           d S )Nr*   r<   r=   c                  <    t          j        d j        d          S )Nr*   rB   r=   )r   get_model_infor   r   s   r   rC   z2TestModelHub.test_get_model_info.<locals>.<lambda>b   s    C$6w	QS$T$T$T r   )r   rU   r   rE   rF   r   s   `r   test_get_model_infoz TestModelHub.test_get_model_info^   sc    7DIQ77777DI...TTTT	
 	
 	
 	
 	
r   c                     t          j        d          }t          j        |          }|                     |t
                     |                     d|d           |                     d|d           d S )Nr*   z
model.onnxzOnnx model not found)member	containermsgtest_data_set_0zTest data not found)r   download_model_with_test_dataoslistdirr   strassertIn)r   	directoryfiless      r   "test_download_model_with_test_dataz/TestModelHub.test_download_model_with_test_datae   s}    5g>>	
9%%i---\U@VWWW$;P 	 	
 	
 	
 	
 	
r   c                 h    t          j        dd          }|                     |t                     d S )NzResNet50-fp32zResNet-preproc)preprocessing_model)r   load_composite_modelr   r   r>   s     r   test_model_with_preprocessingz*TestModelHub.test_model_with_preprocessingn   s=    (1A
 
 
 	eZ00000r   )r
   N)__name__
__module____qualname__r   r(   r2   r4   r:   r?   rG   rL   rR   rV   rc   rg    r   r   r	   r	      s       
   
6 6 6 6K K K K6 6 6 6   1 1 1 1
 
 
 


 
 
 
" " " "
 
 
 

 
 
 
1 1 1 1 1 1r   r	   __main__)r!   r]   unittestos.pathr   pytestonnxr   r   markskipifenvironTestCaser	   rh   mainrk   r   r   <module>rv      s     				                         bj >
:(>$>%   ^1 ^1 ^1 ^1 ^18$ ^1 ^1	 ^1B zHMOOOOO r   