
    h                     $   S SK r S SKrS SKrS SKJr  S SKrS SKJrJr  \R                  R                  S\R                  ;  =(       d    \R                  S   (       + SS9 " S S\R                  5      5       r\S	:X  a  \R                  " 5         gg)
    N)join)
ModelProtohubTEST_HUBzConserving Git LFS quota)reasonc                       \ rS rSrSS jrSS jrSS jrSS jrSS jrSS jr	SS	 jr
SS
 jrSS jrSS jrSS jrSS jrSrg)TestModelHub   Nc                 .    SU l         SU l        SU l        g )NMNISTzonnx/models:main   )namerepoopsetselfs    L/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/test/hub_test.pysetUpTestModelHub.setUp   s    	&	
    c           	      2   [         R                  " U R                  U R                  SS9nU R	                  U[
        5        [        [        R                  " [        [         R                  " 5       SS5      SS95      nU R                  [        U5      S5        g )NT)force_reload***.onnx	recursive   r   loadr   r   assertIsInstancer   listglobr   get_dirassertGreaterEquallenr   modelcached_filess      r   test_force_reloadTestModelHub.test_force_reload   sj    DIIDAeZ0IId3;;=$9TJ
 	L 115r   c                    [         R                  " U R                  SS/S9n[         R                  " U R                  S/S9n[         R                  " U R                  5      nU R                  [	        U5      S5        U R                  [	        U5      [	        U5      5        U R                  [	        U5      [	        U5      5        g )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_models TestModelHub.test_listing_models#   s    OODIIWH:VOODIIXJGOODII63011530137H3IJ30137H3IJr   c           	      4   [         R                  " U R                  U R                  5      nU R	                  U[
        5        [        [        R                  " [        [         R                  " 5       SS5      SS95      nU R                  [        U5      S5        g )Nr   r   Tr   r   r   r&   s      r   test_basic_usageTestModelHub.test_basic_usage,   sh    DII.eZ0IId3;;=$9TJ
 	L 115r   c           	         [         R                  " 5       n[        US5      n[         R                  " U5        [         R                  " U R
                  U R                  5      nU R                  U[        5        [        [        R                  " [        USS5      SS95      nU R                  [        U5      S5        [         R                  " U5        g )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_cacheTestModelHub.test_custom_cache5   s    KKM	H-	IDII.eZ0DIId9dH&EQUVWL 115Ir   c                     [         R                  " U R                  U R                  SS9nU R	                  U[
        5        g )N   r   )r   r   r   r   r    r   r   r'   s     r   test_download_with_opset%TestModelHub.test_download_with_opsetB   s,    DIIQ7eZ0r   c                 :   ^  T R                  [        U 4S j5        g )Nc                  X   > [         R                  " T R                  T R                  SS9$ )NrB   )r   r   r   r   r   s   r   <lambda>/TestModelHub.test_opset_error.<locals>.<lambda>H   s    CHHTYY		$Lr   assertRaisesAssertionErrorr   s   `r   test_opset_errorTestModelHub.test_opset_errorF   s    L	
r   c                 :   ^  T R                  [        U 4S j5        g )Nc                  D   > [         R                  " T R                  SSS9$ )Nonnx/models:unknownT)silent)r   r   r   r   s   r   rI   6TestModelHub.test_manifest_not_found.<locals>.<lambda>N   s    CHHTYY(=dKr   rK   r   s   `r   test_manifest_not_found$TestModelHub.test_manifest_not_foundK   s    K	
r   c                    [         R                  " S5      nU R                  U5        [         R                  " S5      nU R                  U5        [         R                  " U R                  5      nU R	                  U5        g )Nzmhamilton723/modelsrR   )r   _verify_repo_refassertFalser   
assertTrue)r   verifieds     r   test_verify_repo_ref!TestModelHub.test_verify_repo_refQ   sd    ''(=>" ''(=>" ''		2!r   c                    ^  [         R                  " ST R                  SS9  [         R                  " ST R                  5        T R                  [        U 4S j5        g )Nr,   rA   rB   c                  D   > [         R                  " ST R                  SS9$ )Nr,   rH   rB   )r   get_model_infor   r   s   r   rI   2TestModelHub.test_get_model_info.<locals>.<lambda>b   s    C$6$6w		QS$Tr   )r   r`   r   rL   rM   r   s   `r   test_get_model_info TestModelHub.test_get_model_info^   sA    7DIIQ77DII.T	
r   c                     [         R                  " S5      n[        R                  " U5      nU R	                  U[
        5        U R                  SUSS9  U R                  SUSS9  g )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_data/TestModelHub.test_download_model_with_test_datae   s\    55g>	

9%i-\U@VW$;P 	 	
r   c                 Z    [         R                  " SSS9nU R                  U[        5        g )NzResNet50-fp32zResNet-preproc)preprocessing_model)r   load_composite_modelr    r   rC   s     r   test_model_with_preprocessing*TestModelHub.test_model_with_preprocessingn   s*    ((1A
 	eZ0r   )r   r   r   )returnN)__name__
__module____qualname____firstlineno__r   r)   r4   r7   r>   rD   rN   rU   r\   rb   rp   ru   __static_attributes__ r   r   r	   r	      s?    

6K61


"

1r   r	   __main__)r"   rj   unittestos.pathr   pytestonnxr   r   markskipifenvironTestCaser	   rx   mainr}   r   r   <module>r      s     	      bjj >

:(>$>%  ^18$$ ^1	^1B zMMO r   