costs_workflow/tests/unit_tests.py
2023-07-18 11:05:29 -04:00

74 lines
2.9 KiB
Python

import glob
import os
import subprocess
import unittest
from pathlib import Path
from hub.exports.energy_building_exports_factory import EnergyBuildingsExportsFactory
from hub.exports.exports_factory import ExportsFactory
from hub.imports.construction_factory import ConstructionFactory
from hub.imports.energy_systems_factory import EnergySystemsFactory
from hub.imports.geometry_factory import GeometryFactory
from hub.imports.results_factory import ResultFactory
from hub.imports.usage_factory import UsageFactory
from hub.helpers.dictionaries import Dictionaries
from costs.cost import Cost
from costs.constants import SKIN_RETROFIT, SKIN_RETROFIT_AND_SYSTEM_RETROFIT_AND_PV, SYSTEM_RETROFIT_AND_PV
class UnitTests(unittest.TestCase):
def setUp(self) -> None:
city_file = Path("./tests/data/test.geojson").resolve()
output_path = Path('./tests/output/').resolve()
city = GeometryFactory('geojson',
city_file,
height_field='citygml_me',
year_of_construction_field='ANNEE_CONS',
function_field='CODE_UTILI',
function_to_hub=Dictionaries().montreal_function_to_hub_function).city
ConstructionFactory('nrcan', city).enrich()
UsageFactory('nrcan', city).enrich()
ExportsFactory('sra', city, output_path).export()
sra_file = str((output_path / f'{city.name}_sra.xml').resolve())
subprocess.run(['/usr/local/bin/sra', sra_file])
ResultFactory('sra', city, output_path).enrich()
for building in city.buildings:
building.energy_systems_archetype_name = 'system 1 gas pv'
EnergySystemsFactory('montreal_custom', city).enrich()
EnergyBuildingsExportsFactory('insel_monthly_energy_balance', city, output_path).export()
_insel_files = glob.glob(f'{output_path}/*.insel')
for insel_file in _insel_files:
subprocess.run(['insel', str(insel_file)], stdout=subprocess.DEVNULL)
ResultFactory('insel_monthly_energy_balance', city, output_path).enrich()
self._city = city
def test_current_status(self):
for building in self._city.buildings:
result = Cost(building).life_cycle
self.assertIsNotNone(result)
self.assertEqual(0, result.values[0])
def test_scenario_1(self):
for building in self._city.buildings:
result = Cost(building, retrofit_scenario=SKIN_RETROFIT).life_cycle
self.assertIsNotNone(result)
def test_scenario_2(self):
for building in self._city.buildings:
result = Cost(building, retrofit_scenario=SYSTEM_RETROFIT_AND_PV).life_cycle
self.assertIsNotNone(result)
self.assertEqual(0, result.values[0])
def test_scenario_3(self):
for building in self._city.buildings:
result = Cost(building, retrofit_scenario=SKIN_RETROFIT_AND_SYSTEM_RETROFIT_AND_PV).life_cycle
self.assertIsNotNone(result)
def tearDown(self):
files = glob.glob('output/[!.]*')
for file in files:
os.unlink(file)