Fix: List index out of range by teaser_id

districtgenerator/classes/envelope.py

@@ -516,26 +516,26 @@ def loadAreas(self, prj):
 
 
             self.A = {}  # in m2
-            self.A["f"] = prj.buildings[self.id].net_leased_area
+            self.A["f"] = prj.buildings[self.teaser_id].net_leased_area
 
             drct = ("south", "west", "north", "east")
             self.A["opaque"] = {}
-            self.A["opaque"]["south"] = prj.buildings[self.id].outer_area[0.0]
-            self.A["opaque"]["north"] = prj.buildings[self.id].outer_area[180.0]
+            self.A["opaque"]["south"] = prj.buildings[self.teaser_id].outer_area[0.0]
+            self.A["opaque"]["north"] = prj.buildings[self.teaser_id].outer_area[180.0]
             try:
-                self.A["opaque"]["west"] = prj.buildings[self.id].outer_area[90.0]
-                self.A["opaque"]["east"] = prj.buildings[self.id].outer_area[270.0]
+                self.A["opaque"]["west"] = prj.buildings[self.teaser_id].outer_area[90.0]
+                self.A["opaque"]["east"] = prj.buildings[self.teaser_id].outer_area[270.0]
             except KeyError:
                 self.A["opaque"]["west"] = 0.0
                 self.A["opaque"]["east"] = 0.0
 
             try:
-                self.A["opaque"]["roof"] = prj.buildings[self.id].outer_area[-1]
+                self.A["opaque"]["roof"] = prj.buildings[self.teaser_id].outer_area[-1]
             except KeyError:
                 self.A["opaque"]["roof"] = 1.2 * prj.buildings[
-                    self.id].outer_area[-2]
+                    self.teaser_id].outer_area[-2]
 
-            self.A["opaque"]["floor"] = prj.buildings[self.id].outer_area[-2]
+            self.A["opaque"]["floor"] = prj.buildings[self.teaser_id].outer_area[-2]
             self.A["opaque"]["wall"] = sum(self.A["opaque"][d] for d in drct)
 
             # Area of internal floor equals usable area
@@ -546,12 +546,12 @@ def loadAreas(self, prj):
             self.A["opaque"]["intWall"] = 1.5 * self.A["opaque"]["wall"]
 
             self.A["window"] = {}
-            self.A["window"]["south"] = prj.buildings[self.id].window_area[0.0]
-            self.A["window"]["north"] = prj.buildings[self.id].window_area[180.0]
+            self.A["window"]["south"] = prj.buildings[self.teaser_id].window_area[0.0]
+            self.A["window"]["north"] = prj.buildings[self.teaser_id].window_area[180.0]
             try:
-                self.A["window"]["west"] = prj.buildings[self.id].window_area[90.0]
+                self.A["window"]["west"] = prj.buildings[self.teaser_id].window_area[90.0]
                 self.A["window"]["east"] = prj.buildings[
-                    self.id].window_area[270.0]
+                    self.teaser_id].window_area[270.0]
             except KeyError:
                 self.A["window"]["west"] = 0.0
                 self.A["window"]["east"] = 0.0

tests/test_mix_used.py

@@ -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()