|
13 | 13 | class TestDataProcessor(unittest.TestCase): |
14 | 14 | def test_initialization(self): |
15 | 15 | # Arrange |
16 | | - input_folder = os.getcwd() |
17 | | - print(input_folder) |
18 | | - model_name = "test_model" |
| 16 | + test_model_library_path = os.path.abspath( |
| 17 | + os.path.join(os.path.dirname(__file__), "..", "test_model_library") |
| 18 | + ) |
| 19 | + model_name = "dummy" |
19 | 20 | year = 2024 |
20 | 21 | frequency = 50 |
21 | 22 |
|
22 | 23 | # Act |
23 | 24 | processor = DataProcessor( |
24 | | - input_folder=input_folder, |
| 25 | + input_folder=test_model_library_path, |
25 | 26 | model_name=model_name, |
26 | 27 | year=year, |
27 | 28 | frequency=frequency, |
28 | 29 | ) |
29 | 30 |
|
| 31 | + processor.execute_data_pipeline() |
| 32 | + |
30 | 33 | # Assert |
31 | 34 | self.assertEqual(processor.model_name, model_name) |
32 | 35 | self.assertEqual(processor.year, year) |
33 | 36 | self.assertEqual(processor.frequency, frequency) |
34 | 37 | self.assertEqual(processor.wavelength, 6000) |
35 | 38 | self.assertEqual( |
36 | 39 | processor.model_folder, |
37 | | - os.path.join(get_model_dir(), input_folder, model_name), |
| 40 | + os.path.join(get_model_dir(), test_model_library_path, model_name), |
38 | 41 | ) |
39 | 42 | # Timeseries should have 8760 rows |
40 | 43 | self.assertEqual(processor.cycle_map, {}) |
41 | 44 | self.assertEqual(processor.thermal_derate_factors.shape[0], 8760) |
42 | | - self.assertEqual(processor.marginal_costs.shape[0], 8760) |
43 | 45 |
|
44 | 46 |
|
45 | 47 | if __name__ == "__main__": |
|
0 commit comments