Skip to content

Commit d5f62cf

Browse files
committed
Re-add test models
1 parent 76cc2de commit d5f62cf

14 files changed

+61363
-4
lines changed

src/pownet/folder_utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,3 +19,8 @@ def get_database_dir() -> str:
1919
def get_test_dir() -> str:
2020
"""Returns the test directory of the pownet package."""
2121
return os.path.join(get_pownet_dir(), "src", "test_pownet")
22+
23+
24+
def get_test_model_dir() -> str:
25+
"""Returns the test directory of the pownet package."""
26+
return os.path.join(get_pownet_dir(), "src", "test_pownet", "test_model_library")

src/test_pownet/test_core/test_builder.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,23 +3,22 @@
33
import os
44
import unittest
55
from pownet import ModelBuilder, SystemInput
6-
from pownet.folder_utils import get_pownet_dir
76

87
from pownet.builder.thermal import ThermalUnitBuilder
98
from pownet.builder.hydro import HydroUnitBuilder
109
from pownet.builder.nondispatch import NonDispatchUnitBuilder
1110
from pownet.builder.energy_storage import EnergyStorageUnitBuilder
1211
from pownet.builder.system import SystemBuilder
1312

13+
from pownet.folder_utils import get_test_model_dir
14+
1415

1516
class TestModelBuilder(unittest.TestCase):
1617
"""Unless otherwise stated, use "dummy_trade" over 24-hr as the test case."""
1718

1819
def setUp(self) -> None:
1920
# Load the test data
20-
test_model_library_path = os.path.abspath(
21-
os.path.join(get_pownet_dir(), "model_library")
22-
)
21+
test_model_library_path = get_test_model_dir()
2322

2423
self.inputs = SystemInput(
2524
input_folder=test_model_library_path,

0 commit comments

Comments
 (0)