Fix test paths

Author: phumthep

src/test_pownet/test_core/test_data_processor.py

@@ -1,14 +1,11 @@
-""" This is test_data_processor.py
-"""
+"""This is test_data_processor.py"""
 
 import os
 import unittest
 from pownet.core.data_processor import (
     DataProcessor,
 )
 
-from pownet.folder_utils import get_model_dir
-
 
 class TestDataProcessor(unittest.TestCase):
     def test_initialization(self):
@@ -37,7 +34,7 @@ def test_initialization(self):
         self.assertEqual(processor.wavelength, 6000)
         self.assertEqual(
             processor.model_folder,
-            os.path.join(get_model_dir(), test_model_library_path, model_name),
+            os.path.join(test_model_library_path, model_name),
         )
         # Timeseries should have 8760 rows
         self.assertEqual(processor.cycle_map, {})

src/test_pownet/test_model.py

@@ -1,13 +1,10 @@
-""" This is test_model.py
-"""
+"""This is test_model.py"""
 
-import os
 import unittest
-from unittest.mock import patch, MagicMock, call
 
 import gurobipy as gp
 import pandas as pd
-from pownet.modeling import PowerSystemModel
+from pownet.optim_model import PowerSystemModel
 
 
 class TestPowerSystemModel(unittest.TestCase):