From a98cb359d5f393c255b6917e4c55b1ede831a5c9 Mon Sep 17 00:00:00 2001 From: RaulDeMaio Date: Sat, 24 May 2025 17:34:35 +0200 Subject: [PATCH] Fix package import for tests --- {{cookiecutter.directory_name}}/src/__init__.py | 1 + {{cookiecutter.directory_name}}/src/main.py | 2 +- {{cookiecutter.directory_name}}/tests/test_main.py | 9 +++++++++ 3 files changed, 11 insertions(+), 1 deletion(-) create mode 100644 {{cookiecutter.directory_name}}/src/__init__.py diff --git a/{{cookiecutter.directory_name}}/src/__init__.py b/{{cookiecutter.directory_name}}/src/__init__.py new file mode 100644 index 0000000..8b13789 --- /dev/null +++ b/{{cookiecutter.directory_name}}/src/__init__.py @@ -0,0 +1 @@ + diff --git a/{{cookiecutter.directory_name}}/src/main.py b/{{cookiecutter.directory_name}}/src/main.py index 83d2334..f67bd8b 100644 --- a/{{cookiecutter.directory_name}}/src/main.py +++ b/{{cookiecutter.directory_name}}/src/main.py @@ -1,4 +1,4 @@ -from logging_utils import setup_logger +from .logging_utils import setup_logger log = setup_logger() diff --git a/{{cookiecutter.directory_name}}/tests/test_main.py b/{{cookiecutter.directory_name}}/tests/test_main.py index 77155fa..d3f8b25 100644 --- a/{{cookiecutter.directory_name}}/tests/test_main.py +++ b/{{cookiecutter.directory_name}}/tests/test_main.py @@ -1,4 +1,13 @@ # tests/test_main.py +from pathlib import Path +import sys + +# Ensure src is importable whether tests are run from the project root or template root +PROJECT_ROOT = Path(__file__).resolve().parents[1] +if str(PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(PROJECT_ROOT)) + + def test_main_runs(): from src.main import main assert main() is None # or your actual expected behavior