
    wi1                       d dl mZ 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 )    )annotationsN)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returnNonec                0    d| _         d| _        d| _        d S )NMNISTzonnx/models:main   )namerepoopsetselfs    b/root/.openclaw/workspace/chatterbox_venv_py311/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   rE   z6TestModelHub.test_manifest_not_found.<locals>.<lambda>N   s    CHTY(=dKKK r   rF   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/modelsrL   )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,   rD   r?   )r   get_model_infor   r   s   r   rE   z2TestModelHub.test_get_model_info.<locals>.<lambda>b   s    C$6w	QS$T$T$T r   )r   rW   r   rG   rH   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   N)r   r   )__name__
__module____qualname__r   r*   r4   r6   r<   rA   rI   rN   rT   rX   re   ri    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__)
__future__r   r#   r_   unittestos.pathr   pytestonnxr   r   markskipifenvironTestCaser
   rj   mainrm   r   r   <module>ry      s   # " " " " "  				                         bj >
:(>$>%   ^1 ^1 ^1 ^1 ^18$ ^1 ^1	 ^1B zHMOOOOO r   