o
    Ďi                     @   sL   d dl Z d dlmZ d dlmZmZ G dd de jZG dd de jZdS )    N)import_module)get_model_formatget_wrong_dependencies_versionsc                   @      e Zd Zdd ZdS )TestVersionsc           	      C   sf   d}t |j}dd|fdddddg}g d}t||D ]\}}t|g}t|d	k}| || qd S )
Ntests.util.dummy_modulez==)r   <=z0.2.0)r   r   z1.0.0)r   >=z0.1.0)r   r	   z0.6.0)r   r	   z0.1.34)TFTTFTr   )r   __version__zipr   lenassertEqual)	selfmodule_pathmodule_versiontestsexpected_resultstestexpected_resultwrong_dependencies_versionsis_correct_dep r   T/home/jeff/fluffinator/venv/lib/python3.10/site-packages/tests/util/test_versions.py test_wrong_dependencies_versions   s   
	
z-TestVersions.test_wrong_dependencies_versionsN)__name__
__module____qualname__r   r   r   r   r   r          r   c                   @   r   )TestGetModelFormatc              	   C   sV   g d}|D ]"\}}| j |d | t|| W d    n1 s#w   Y  qd S )N))	yolov5v2syolov5pytorch)yolov11nr    )zrf-detr-nas-parentcoco)zrfdetr-nanor"   )zvit-base-patch16-224-in21kfolder)resnet14r#   )	resenet38r    )zinvlid-typer    )
model_type)subTestr   r   )r   casesr&   expected_formatr   r   r   &test_get_model_format_with_various_ids   s   z9TestGetModelFormat.test_get_model_format_with_various_idsN)r   r   r   r*   r   r   r   r   r      r   r   )	unittest	importlibr   roboflow.util.versionsr   r   TestCaser   r   r   r   r   r   <module>   s
    