def test_import_hydx_unknown_connection_types(caplog):
hydx = mock.Mock()
hydx.connection_nodes = []
hydx.structures = []
hydx.surfaces = []
hydx.discharges = []
hydx.connections = [
mock.Mock(identificatieknooppuntofverbinding="ovs82", typeverbinding="XXX")
]
threedi = Threedi()
threedi.import_hydx(hydx)
assert '"typeverbinding" is not recognized' in caplog.text
def test_structure_does_not_exist_error(caplog):
hydx = mock.Mock()
hydx.structures = []
hydx.connection_nodes = []
hydx.surfaces = []
hydx.discharges = []
hydx.connections = [
mock.Mock(identificatieknooppuntofverbinding="pmp1", typeverbinding="PMP")
]
threedi = Threedi()
threedi.import_hydx(hydx)
assert "Structure does not exist for connection" in caplog.text
def test_get_mapping_value_right(caplog):
MANHOLE_SHAPE_MAPPING = {
"RND": Constants.MANHOLE_SHAPE_ROUND,
"RHK": Constants.MANHOLE_SHAPE_RECTANGLE,
}
shape_code = "RND"
record_code = "01_TEST"
threedi = Threedi()
shape = threedi.get_mapping_value(
MANHOLE_SHAPE_MAPPING, shape_code, record_code, name_for_logging="manhole shape"
)
assert shape == "rnd"
def test_get_mapping_value_wrong(caplog):
MANHOLE_SHAPE_MAPPING = {
"RND": Constants.MANHOLE_SHAPE_ROUND,
"RHK": Constants.MANHOLE_SHAPE_RECTANGLE,
}
shape_code = "SQR"
record_code = "01_TEST"
threedi = Threedi()
threedi.get_mapping_value(
MANHOLE_SHAPE_MAPPING, shape_code, record_code, name_for_logging="manhole shape"
)
assert "Unknown" in caplog.text
def setUp(self):
self.threedi = Threedi()
hydx_path = "hydxlib/tests/example_files_structures_hydx/"
self.threedi_db_settings = {
"threedi_dbname": "test_gwsw",
"threedi_host": "localhost",
"threedi_user": "******",
"threedi_password": "******",
"threedi_port": 5432,
}
self.hydx = import_hydx(hydx_path)
self.threedi.import_hydx(self.hydx)
class TestThreedi(TestCase):
def setUp(self):
self.threedi = Threedi()
hydx_path = "hydxlib/tests/example_files_structures_hydx/"
self.threedi_db_settings = {
"threedi_dbname": "test_gwsw",
"threedi_host": "localhost",
"threedi_user": "******",
"threedi_password": "******",
"threedi_port": 5432,
}
self.hydx = import_hydx(hydx_path)
self.threedi.import_hydx(self.hydx)
@pytest.fixture(autouse=True)
def inject_fixtures(self, caplog):
self._caplog = caplog
def test_export_threedi(self):
output = export_threedi(self.hydx, self.threedi_db_settings)
assert len(output.connection_nodes) == 85
def test_write_to_db_con_nodes_huge(self):
commit_counts_expected = {
"connection_nodes": 85,
"manholes": 84,
"pumpstations": 8,
"weirs": 6,
"cross_sections": 38,
"orifices": 2,
"impervious_surfaces": 330,
"pipes": 80,
"outlets": 3,
}
commit_counts = write_threedi_to_db(self.threedi,
self.threedi_db_settings)
assert commit_counts == commit_counts_expected
def setUp(self):
self.threedi = Threedi()
hydx_path = "hydxlib/tests/example_files_structures_hydx/"
self.hydx = import_hydx(hydx_path)
class TestThreedi(TestCase):
def setUp(self):
self.threedi = Threedi()
hydx_path = "hydxlib/tests/example_files_structures_hydx/"
self.hydx = import_hydx(hydx_path)
@pytest.fixture(autouse=True)
def inject_fixtures(self, caplog):
self._caplog = caplog
def test_import_hydx(self):
self.threedi.import_hydx(self.hydx)
assert len(self.threedi.connection_nodes) == 85
def test_add_connection_node(self):
connection_node_0 = {
"code": "knp1",
"initial_waterlevel": None,
"geom": (400, 50, 28992),
}
self.threedi.import_hydx(self.hydx)
assert self.threedi.connection_nodes[0] == connection_node_0
def test_add_connection_node_manhole(self):
manhole_0 = {
"code": "knp1",
"display_name": "1001",
"surface_level": 2.75,
"width": 7071,
"length": 7071,
"shape": "rnd",
"bottom_level": 0,
"calculation_type": 2,
"manhole_indicator": 0,
}
self.threedi.import_hydx(self.hydx)
assert self.threedi.manholes[0] == manhole_0
def test_add_pumpstation(self):
pumpstation_1 = {
"code": "pmp88",
# check if connection number 1 is created for second structure with these nodes
"display_name": "2001-1016-1",
"start_node.code": "knp72",
"end_node.code": "knp15",
"type_": 1,
"start_level": 0.5,
"lower_stop_level": 0,
"upper_stop_level": None,
"capacity": 20,
"sewerage": True,
}
self.threedi.import_hydx(self.hydx)
assert self.threedi.pumpstations[0] == pumpstation_1
def test_add_pipe(self):
pipe_1 = {
"code": "lei1",
# check if connection number 1 is created for second structure with these nodes
"display_name": "1003-1004-1",
"start_node.code": "knp3",
"end_node.code": "knp4",
"sewerage_type": 0,
"invert_level_start_point": 0.10,
"invert_level_end_point": 0.00,
"original_length": 48.0,
"material": 0,
"sewerage_type": 0,
"calculation_type": 1,
"cross_section_details": {
"code": "round_1.1",
"shape": 2,
"width": 1.1,
"height": None,
},
"cross_section_code": "round_1.1",
}
self.threedi.import_hydx(self.hydx)
assert self.threedi.pipes[0] == pipe_1
def test_add_surface(self):
surface_1 = {
"code": "1",
"display_name": "knp8",
"area": 9.0,
"surface_class": "gesloten verharding",
"surface_inclination": "uitgestrekt",
}
self.threedi.import_hydx(self.hydx)
assert self.threedi.impervious_surfaces[0] == surface_1
def test_add_discharge(self):
discharge_3 = {
"code": "263",
"display_name": "knp61",
"area": 0.0,
"surface_class": "gesloten verharding",
"surface_inclination": "vlak",
"dry_weather_flow": 60.0,
"nr_of_inhabitants": "2",
}
self.threedi.import_hydx(self.hydx)
assert self.threedi.impervious_surfaces[262] == discharge_3
def test_add_boundary(self):
boundary_1 = {
"node.code": "knp78",
"timeseries": "0,-5.0\n9999,-5.0 ",
"boundary_type": 1,
}
self.threedi.import_hydx(self.hydx)
assert self.threedi.outlets[0] == boundary_1
# def test_add_first_pump_with_same_code(self):
# self.threedi.import_hydx(self.hydx)
# # select first manhole from dataset for check
# connection = self.hydx.connections[82]
# structure = self.hydx.structures[5]
# self.threedi.add_structure(connection, structure)
# assert "Only first structure" in self._caplog.text
def test_add_pump_type_2(self):
self.threedi.import_hydx(self.hydx)
# select first manhole from dataset for check
connection = self.hydx.connections[90]
structure = self.hydx.structures[13]
self.threedi.add_structure(connection, structure)
assert self.threedi.pumpstations[8]["type_"] == 2
def test_add_weir_with_boundary_and_open_rectangle_profile(self):
weir_1 = {
"code": "ovs83",
"display_name": "1009-1009-1",
"start_node.code": "knp8",
"end_node.code": "knp55",
"cross_section_details": {
"shape": 1,
"width": 3,
"height": None,
},
"crest_type": 4,
"crest_level": 2.7,
"discharge_coefficient_positive": 0.9,
"discharge_coefficient_negative": 0.9,
"sewerage": True,
"cross_section_code": "default",
}
self.threedi.import_hydx(self.hydx)
assert self.threedi.weirs[1] == weir_1
def test_add_orifice_with_round_profile(self):
orifice_1 = {
"code": "drl97",
"display_name": "2002-2002-1",
"start_node.code": "knp16",
"end_node.code": "knp60",
"cross_section_details": {"shape": 2, "width": 0.4, "height": None},
"discharge_coefficient_positive": 0,
"discharge_coefficient_negative": 0.8,
"sewerage": True,
"crest_type": 4,
"crest_level": 0.0,
"cross_section_code": "round_0.4",
}
self.threedi.import_hydx(self.hydx)
assert self.threedi.orifices[1] == orifice_1
def export_threedi(hydx, threedi_db_settings):
threedi = Threedi()
threedi.import_hydx(hydx)
commit_counts = write_threedi_to_db(threedi, threedi_db_settings)
logger.info("GWSW-hydx exchange created elements: %r", commit_counts)
return threedi