Tests to incorporate HVAC systems to EP workflow

main.py

@@ -12,7 +12,7 @@ from sra_engine import SraEngine
 
 
 try:
-  file_path = (Path(__file__).parent / 'input_files' / '228730.geojson')
+  file_path = (Path(__file__).parent / 'input_files' / 'summerschool_all_buildings.geojson')
   construction_format = 'nrcan'
   usage_format = 'nrcan'
   energy_systems_format = 'montreal_custom'
@@ -23,7 +23,7 @@ try:
   print('[simulation start]')
   city = GeometryFactory('geojson',
                          path=file_path,
-                         height_field='heightmax',
+                         height_field='citygml_me',
                          year_of_construction_field='ANNEE_CONS',
                          function_field='CODE_UTILI',
                          function_to_hub=Dictionaries().montreal_function_to_hub_function).city
@@ -33,8 +33,14 @@ try:
   print('enrich constructions... done')
   UsageFactory(usage_format, city).enrich()
   print('enrich usage... done')
+  i = 1
+  x = len(city.buildings)
   for building in city.buildings:
-    building.energy_systems_archetype_name = 'system 3 and 4 electricity'
+    if i < x:
+      building.energy_systems_archetype_name = 'system 1 gas'
+    else:
+      building.energy_systems_archetype_name = 'system 6 gas'
+    i = i + 1
   EnergySystemsFactory(energy_systems_format, city).enrich()
   print('enrich systems... done')