Fix unit test issues

Author: phumthep

src/test_pownet/test_core/test_builder.py

@@ -1,14 +1,10 @@
 """ Unit tests for the ModelBuilder class.
 """
 
+import os
 import unittest
-from unittest.mock import MagicMock, patch
-
-import gurobipy as gp
-
-from pownet.folder_utils import get_output_dir, get_model_dir
-from pownet.modeling import PowerSystemModel
 from pownet.core import ModelBuilder, SystemInput
+from pownet.folder_utils import get_model_dir
 
 
 class TestModelBuilder(unittest.TestCase):
@@ -17,7 +13,8 @@ class TestModelBuilder(unittest.TestCase):
     def setUp(self) -> None:
         # Load the test data
         self.inputs = SystemInput(
-            model_name="dummy_trade",
+            input_folder=get_model_dir(),
+            model_name="dummy",
             year=2016,
             sim_horizon=24,
         )

src/test_pownet/test_core/test_data_processor.py

@@ -3,24 +3,29 @@
 
 import os
 import unittest
-from unittest.mock import patch, MagicMock
 from pownet.core.data_processor import (
     DataProcessor,
 )
 
-from pownet.data_utils import get_dates
-from pownet.folder_utils import get_database_dir, get_model_dir
+from pownet.folder_utils import get_model_dir
 
 
 class TestDataProcessor(unittest.TestCase):
     def test_initialization(self):
         # Arrange
+        input_folder = os.getcwd()
+        print(input_folder)
         model_name = "test_model"
         year = 2024
         frequency = 50
 
         # Act
-        processor = DataProcessor(model_name, year, frequency)
+        processor = DataProcessor(
+            input_folder=input_folder,
+            model_name=model_name,
+            year=year,
+            frequency=frequency,
+        )
 
         # Assert
         self.assertEqual(processor.model_name, model_name)
@@ -29,7 +34,7 @@ def test_initialization(self):
         self.assertEqual(processor.wavelength, 6000)
         self.assertEqual(
             processor.model_folder,
-            os.path.join(get_model_dir(), model_name),
+            os.path.join(get_model_dir(), input_folder, model_name),
         )
         # Timeseries should have 8760 rows
         self.assertEqual(processor.cycle_map, {})