Hopefully pipeline passes

beeldengeluid · Oct 8, 2024 · b0c753a · b0c753a
1 parent 4fcf121
commit b0c753a
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 9 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,6 +15,7 @@ boto3 = "^1.35.10"
 fastapi = "^0.115.0"
 uvicorn = "^0.30.6"
 tomli = "^2.0.1"
+python-dotenv = "^1.0.1"
 
 [tool.poetry.group.dev.dependencies]
 moto = "^5.0.13"

diff --git a/tests/.env b/tests/.env
@@ -0,0 +1,3 @@
+DATA_BASE_DIR=data
+MODEL_BASE_DIR=model
+S3_ENDPOINT_URL=https://someurl.org
diff --git a/tests/test_model_download.py b/tests/test_model_download.py
@@ -1,19 +1,14 @@
 import pytest
 import shutil
 import os
-import logging
+from dotenv import load_dotenv
 
-# Mocking environment used in model_download
-os.environ["DATA_BASE_DIR"] = "data"
-os.environ["MODEL_BASE_DIR"] = "model"
-os.environ["S3_ENDPOINT_URL"] = "https://someurl.org"
+# Mocking environment used in model_download (by loading tests/.env)
+load_dotenv()
 
 from model_download import extract_model, get_model_location  # noqa
 
 
-logger = logger = logging.getLogger(__name__)
-
-
 @pytest.mark.parametrize(
     "destination, extension, expected_output",
     [