Adapt tests to package directory structure.

Signed-off-by: Bobby Noelte <b0661n0e17e@gmail.com>

Makefile

@@ -1,5 +1,5 @@
 # Define the targets
-.PHONY: help venv pip docker-run docs clean
+.PHONY: help venv pip test docker-run docs clean
 
 # Default target
 all: help
@@ -38,6 +38,11 @@ run:
 	@echo "Starting flask server, please wait..."
 	.venv/bin/python ./flask_server.py
 
+# Target to run tests.
+test:
+	@echo "Running tests..."
+	.venv/bin/pytest
+
 # Run entire setup on docker
 docker-run:
 	@docker compose up

pyproject.toml

@@ -5,3 +5,10 @@ profile = "black"
 ignore = [
     "F841",  # don't complain about unused variables
 ]
+
+[tool.pytest.ini_options]
+minversion = "7.0"
+addopts = "--cov=src"
+pythonpath = [ "src", ]
+testpaths = [ "tests", ]
+

tests/test_class_akku.py

@@ -1,6 +1,6 @@
 import unittest
 
-from modules.class_akku import PVAkku
+from akkudoktoreos.class_akku import PVAkku
 
 
 class TestPVAkku(unittest.TestCase):

tests/test_class_ems.py

@@ -1,9 +1,9 @@
 import pytest
 
-from modules.class_akku import PVAkku
-from modules.class_ems import EnergieManagementSystem
-from modules.class_haushaltsgeraet import Haushaltsgeraet
-from modules.class_inverter import Wechselrichter  # Example import
+from akkudoktoreos.class_akku import PVAkku
+from akkudoktoreos.class_ems import EnergieManagementSystem
+from akkudoktoreos.class_haushaltsgeraet import Haushaltsgeraet
+from akkudoktoreos.class_inverter import Wechselrichter  # Example import
 
 prediction_hours = 48
 optimization_hours = 24

tests/test_class_optimize.py

@@ -1,7 +1,7 @@
 import numpy as np
 import pytest
 
-from modules.class_optimize import optimization_problem
+from akkudoktoreos.class_optimize import optimization_problem
 
 # Sample known result (replace with the actual expected output)
 EXPECTED_RESULT = {

tests/test_heatpump.py

@@ -1,6 +1,6 @@
 import pytest
 
-from modules.heatpump import Heatpump
+from akkudoktoreos.heatpump import Heatpump
 
 
 @pytest.fixture(scope="function")