diff --git a/tests/data/test_pvlive_data.py b/tests/data/test_pvlive_data.py index 7a2265e..fbcc9c3 100644 --- a/tests/data/test_pvlive_data.py +++ b/tests/data/test_pvlive_data.py @@ -2,7 +2,7 @@ from unittest.mock import MagicMock from datetime import datetime import pytz -from open_data_pvnet.scripts.fetch_pvlive_data import PVLiveData +from src.open_data_pvnet.scripts.fetch_pvlive_data import PVLiveData @pytest.fixture diff --git a/tests/test_dwd.py b/tests/test_dwd.py index baf9293..693be74 100644 --- a/tests/test_dwd.py +++ b/tests/test_dwd.py @@ -1,7 +1,11 @@ import pytest from unittest.mock import Mock from pathlib import Path +import sys +import pathlib +src_dir = pathlib.Path(__file__).parent.parent / 'src' +sys.path.insert(0, str(src_dir)) from open_data_pvnet.nwp.dwd import ( generate_variable_url, fetch_dwd_data, diff --git a/tests/test_main.py b/tests/test_main.py index cf42ff4..9f1437d 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1,5 +1,11 @@ import pytest from unittest.mock import patch +import sys +import pathlib + +src_dir = pathlib.Path(__file__).parent.parent / 'src' +sys.path.insert(0, str(src_dir)) + from open_data_pvnet.main import ( configure_parser, load_env_and_setup_logger, diff --git a/tests/test_met_office.py b/tests/test_met_office.py index 1f9250e..f59c686 100644 --- a/tests/test_met_office.py +++ b/tests/test_met_office.py @@ -1,5 +1,10 @@ import pytest from unittest.mock import Mock +import sys +import pathlib + +src_dir = pathlib.Path(__file__).parent.parent / 'src' +sys.path.insert(0, str(src_dir)) from open_data_pvnet.nwp.met_office import ( generate_prefix, diff --git a/tests/test_metadata.py b/tests/test_metadata.py index 9e781a4..5104358 100644 --- a/tests/test_metadata.py +++ b/tests/test_metadata.py @@ -1,4 +1,10 @@ import tomllib +import sys +import pathlib + +src_dir = pathlib.Path(__file__).parent.parent / 'src' +sys.path.insert(0, str(src_dir)) + import open_data_pvnet diff --git a/tests/test_utils.py b/tests/test_utils.py index 128dd79..0aca8e2 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -8,7 +8,11 @@ import yaml import pandas as pd from huggingface_hub.utils import EntryNotFoundError +import sys +import pathlib +src_dir = pathlib.Path(__file__).parent.parent / 'src' +sys.path.insert(0, str(src_dir)) from open_data_pvnet.utils.config_loader import load_config from open_data_pvnet.utils.env_loader import load_environment_variables from open_data_pvnet.utils.data_converters import convert_nc_to_zarr