forked from s_ranjbar/city_retrofit
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
"""
|
|
TestSystemsCatalog
|
|
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
|
Copyright © 2022 Concordia CERC group
|
|
Project Coder Pilar Monsalvete Alvarez de Uribarri pilar.monsalvete@concordia.ca
|
|
"""
|
|
|
|
from unittest import TestCase
|
|
from hub.catalog_factories.energy_systems_catalog_factory import EnergySystemsCatalogFactory
|
|
|
|
|
|
class TestSystemsCatalog(TestCase):
|
|
|
|
def test_montreal_custom_catalog(self):
|
|
catalog = EnergySystemsCatalogFactory('montreal_custom').catalog
|
|
|
|
catalog_categories = catalog.names()
|
|
archetypes = catalog.names('archetypes')
|
|
self.assertEqual(23, len(archetypes['archetypes']))
|
|
systems = catalog.names('systems')
|
|
self.assertEqual(18, len(systems['systems']))
|
|
generation_equipments = catalog.names('generation_equipments')
|
|
self.assertEqual(7, len(generation_equipments['generation_equipments']))
|
|
distribution_equipments = catalog.names('distribution_equipments')
|
|
self.assertEqual(8, len(distribution_equipments['distribution_equipments']))
|
|
emission_equipments = catalog.names('emission_equipments')
|
|
self.assertEqual(3, len(emission_equipments['emission_equipments']))
|
|
with self.assertRaises(ValueError):
|
|
catalog.names('unknown')
|
|
|
|
# retrieving all the entries should not raise any exceptions
|
|
for category in catalog_categories:
|
|
for value in catalog_categories[category]:
|
|
catalog.get_entry(value)
|
|
|
|
print(catalog.entries())
|
|
|
|
with self.assertRaises(IndexError):
|
|
catalog.get_entry('unknown')
|
|
|
|
def test_north_america_systems_catalog(self):
|
|
catalog = EnergySystemsCatalogFactory('north_america').catalog
|
|
print(catalog.entries())
|