diff --git a/requirements.txt b/requirements.txt index dc00483..36f99c6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -123,6 +123,7 @@ pymdown-extensions==10.7.1 pyparsing==3.1.1 PySocks==1.7.1 pytest==8.1.1 +python-dotenv==1.0.1 python-dateutil==2.8.2 pytz==2023.3.post1 PyYAML==6.0.1 diff --git a/tests/pytest.ini b/tests/pytest.ini deleted file mode 100644 index 8d38fa0..0000000 --- a/tests/pytest.ini +++ /dev/null @@ -1,6 +0,0 @@ -[pytest] - -API_URL = $API_URL -DAGSHUB_REPO_OWNER = $DAGSHUB_REPO_OWNER -DAGSHUB_REPO = $DAGSHUB_REPO -MODEL_URI = $MODEL_URI \ No newline at end of file diff --git a/tests/test_model.py b/tests/test_model.py index a3c0346..123f61c 100644 --- a/tests/test_model.py +++ b/tests/test_model.py @@ -7,21 +7,35 @@ import os +from dotenv import load_dotenv -class TestLoadModel: + +load_dotenv() + + +class TestEnvVar: def test_get_envvar(self): - print("Valeur de 'API_URL' dans l'environnement:", os.getenv('API_URL')) - print("Valeur de 'DAGSHUB_REPO_OWNER' dans l'environnement:", os.getenv('DAGSHUB_REPO_OWNER')) - print("Valeur de 'DAGSHUB_REPO' dans l'environnement:", os.getenv('DAGSHUB_REPO')) - print("Valeur de 'MODEL_URI' dans l'environnement:", os.getenv('MODEL_URI')) + api_url = os.getenv('API_URL') + dagshub_repo_owner = os.getenv('DAGSHUB_REPO_OWNER') + dagshub_repo = os.getenv('DAGSHUB_REPO') + model_uri = os.getenv('MODEL_URI') + + print(f"API URL: {api_url}") + print(f"Dagshub Repo Owner: {dagshub_repo_owner}") + print(f"Dagshub Repo: {dagshub_repo}") + print(f"Model URI: {model_uri}") + + assert api_url is not None + assert dagshub_repo_owner is not None + assert dagshub_repo is not None + assert model_uri is not None + - assert os.getenv('API_URL') is not None - assert os.getenv('DAGSHUB_REPO_OWNER') is not None - assert os.getenv('DAGSHUB_REPO') is not None - assert os.getenv('MODEL_URI') is not None +# class TestLoadModel: + # def test_load_model(self): # model = load_model() # assert model is not None