From 630b36d958f49f8512e5db75720d199e8d2b7173 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Charley=20=E2=88=86=2E=20L=2E?=
 <CharleyDL@users.noreply.github.com>
Date: Thu, 2 May 2024 14:34:34 +0200
Subject: [PATCH] test get env

---
 tests/pytest.ini    |  6 ++++++
 tests/test_model.py | 23 +++++++++--------------
 2 files changed, 15 insertions(+), 14 deletions(-)
 create mode 100644 tests/pytest.ini

diff --git a/tests/pytest.ini b/tests/pytest.ini
new file mode 100644
index 0000000..8d38fa0
--- /dev/null
+++ b/tests/pytest.ini
@@ -0,0 +1,6 @@
+[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 cbdaa19..a3c0346 100644
--- a/tests/test_model.py
+++ b/tests/test_model.py
@@ -7,24 +7,19 @@
 
 import os
 
-# from utils.functions import load_model
-
-
-API_URL = os.environ("API_URL",)
-DAGSHUB_REPO_OWNER = os.environ("DAGSHUB_REPO_OWNER")
-DAGSHUB_REPO = os.environ("DAGSHUB_REPO")
-MODEL_URI = os.environ("MODEL_URI")
-
-
 
 class TestLoadModel:
 
     def test_get_envvar(self):
-        envvar = [API_URL, DAGSHUB_REPO_OWNER, DAGSHUB_REPO, MODEL_URI]
-
-        for var in envvar:
-            assert var in os.environ, f"{var} is not in os.environ"
-
+        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'))
+
+        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
 
 
 #     def test_load_model(self):