forked from RWTH-EBC/districtgenerator
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #38 from c0nb4/37-fix-mix-used
Fix: List index out of range by teaser_id
- Loading branch information
Showing
2 changed files
with
134 additions
and
12 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,122 @@ | ||
| import unittest | ||
| import random as rd | ||
| from random import sample | ||
| import os | ||
| import pandas as pd | ||
| from districtgenerator.classes import datahandler | ||
| from districtgenerator.functions import path_checks | ||
| # Shoud create simulations for non-residential buildings | ||
| # Write a test, that all functions run through without errors | ||
| # Check if the results exist and are not zero | ||
|
|
||
|
|
||
|
|
||
| PARENT_FOLDER_PATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) | ||
|
|
||
| class TextMixUsed(unittest.TestCase): | ||
| def setUp(self): | ||
| # Create test scenario | ||
| self.test_data = { | ||
| 'id': [0, 1, 2, 3], | ||
| 'building': [ | ||
| "IWU Office, Administrative or Government Buildings", | ||
| "IWU Research and University Teaching", | ||
| "SFH", "TH" ], | ||
|
|
||
| 'year': [rd.randint(1900, 2024) for _ in range(4)], | ||
| 'retrofit': [0] * 4, | ||
| 'area': [rd.randint(0, 100) for _ in range(4)], | ||
| 'construction_type': [""] * 4, | ||
| 'night_setback': [rd.randint(0, 1) for _ in range(4)], | ||
| 'heater': ["HP"] * 4, | ||
| 'PV': [rd.randint(0, 1) for _ in range(4)], | ||
| 'STC': [rd.randint(0, 1) for _ in range(4)], | ||
| 'EV': [rd.randint(0, 1) for _ in range(4)], | ||
| 'BAT': [rd.randint(0, 1) for _ in range(4)], | ||
| 'f_TES': [rd.randint(0, 1) for _ in range(4)], | ||
| 'f_BAT': [rd.randint(0, 1) for _ in range(4)], | ||
| 'f_EV': ["M"] * 4, | ||
| 'f_PV': [rd.randint(0, 1) for _ in range(4)], | ||
| 'f_STC': [rd.randint(0, 1) for _ in range(4)], | ||
| 'gamma_PV': [rd.randint(0, 1) for _ in range(4)], | ||
| 'ev_charging': ["on_demand"] * 4 | ||
| } | ||
|
|
||
| df = pd.DataFrame(self.test_data, columns=self.test_data.keys()) | ||
| self.scenario_path = os.path.join(PARENT_FOLDER_PATH, 'districtgenerator', 'data', 'scenarios', 'test_scenario.csv') | ||
| df.to_csv(self.scenario_path, index=False, sep=';') | ||
| df.to_csv(self.scenario_path, index=False, sep=';') | ||
| print(df.head()) | ||
| self.results_path = os.path.join(PARENT_FOLDER_PATH, 'test_scenario', 'demands') | ||
|
|
||
| def test_mix_used(self): | ||
| # Test all functions in the non_residential module | ||
| data = datahandler.Datahandler() | ||
| data.setWeatherFile(os.path.join(PARENT_FOLDER_PATH, 'districtgenerator', 'data', 'weather', 'EPW', 'DEU_BE_Berlin-Schonefeld.AP.103850_TMYx.2004-2018.epw')) | ||
| data.generateEnvironment(plz="52070") | ||
|
|
||
| # Use the created scenario for testing | ||
| data.initializeBuildings('test_scenario') | ||
| data.setResultPath('test_scenario') | ||
| data.generateBuildings() | ||
|
|
||
| data.generateDemands() | ||
|
|
||
| # Add assertions to check if the results exist and are not zero | ||
| self.assertTrue(os.path.exists(self.scenario_path), f"Scenario file not found: {self.scenario_path}") | ||
| self.assertGreater(os.path.getsize(self.scenario_path), 0, f"Scenario file is empty: {self.scenario_path}") | ||
| scenario_data = pd.read_csv(self.scenario_path, sep=';') | ||
| for file in os.listdir(self.results_path): | ||
| if file.endswith('.csv'): | ||
| file_path = os.path.join(self.results_path, file) | ||
| df = pd.read_csv(file_path, sep=',') | ||
| # Check if the required columns are present | ||
| required_columns = ["elec", "dhw", "occ", "gains", "heat"] | ||
| for column in required_columns: | ||
| self.assertIn(column, df.columns, f"Column {column} not found in file: {file_path}") | ||
| self.assertGreaterEqual(df["heat"].sum(), 0, f"Heat sum is negative in file: {file_path}") | ||
| self.assertGreaterEqual(df["elec"].sum(), 0, f"Electricity sum is negative in file: {file_path}") | ||
| self.assertGreaterEqual(df["dhw"].sum(), 0, f"DHW sum is negative in file: {file_path}") | ||
| self.assertGreaterEqual(df["occ"].sum(), 0, f"Occupancy sum is negative in file: {file_path}") | ||
| # Gains can be negative due to cooling | ||
| self.assertTrue(df["gains"].notna().any(), f"Gains column has no values in file: {file_path}") | ||
|
|
||
| scenario_data = pd.read_csv(self.scenario_path, sep=';') | ||
| for index, row in scenario_data.iterrows(): | ||
| building = row['building'] | ||
| print(building) | ||
|
|
||
| building_id = str(index) | ||
| building_file = None | ||
| for file in os.listdir(self.results_path): | ||
| if file.startswith(f"{building_id}_{building}") and file.endswith(".csv"): | ||
| building_file = os.path.join(self.results_path, file) | ||
| df = pd.read_csv(building_file, sep=',') | ||
| energy_demand = df['heat'].sum() / 1000 | ||
| area = row['area'] | ||
| energy_demand_per_square_meter = energy_demand / area | ||
| print(f"Energy demand per square meter for building {building}: {energy_demand_per_square_meter}") | ||
| self.assertGreaterEqual(energy_demand_per_square_meter, 0, f"Energy demand per square meter is negative for building {building}: {energy_demand_per_square_meter}") | ||
| self.assertIsNotNone(building_file, f"No results file found for building {building_id}") | ||
|
|
||
| def tearDown(self): | ||
| if os.path.exists(self.scenario_path): | ||
| os.remove(self.scenario_path) | ||
| if os.path.exists(self.results_path): | ||
| for file in os.listdir(self.results_path): | ||
| file_path = os.path.join(self.results_path, file) | ||
| if os.path.isfile(file_path): | ||
| try: | ||
| os.remove(file_path) | ||
| except PermissionError: | ||
| print(f"Unable to delete {file_path}. File may be in use.") | ||
| try: | ||
| os.rmdir(self.results_path) | ||
| except OSError: | ||
| print(f"Unable to remove directory {self.results_path}. It may not be empty or you may not have permission.") | ||
| else: | ||
| print(f"Results directory not found: {self.results_path}") | ||
|
|
||
|
|
||
| if __name__ == '__main__': | ||
| unittest.main() |