def test_water_body_info(self):
cleaner = SiteNodeGeometryCleaner(self.mainsite)
cleaner.process_level_curves()
cleaner.process_material_areas()
water_shape, water_info = cleaner['{Water body}']
self.assertEqual(water_info["altitude"], self.altitude_water)
def test_export_subsite_feature(self):
cleaner_mainsite = SiteNodeGeometryCleaner(self.mainsite)
cleaner_subsite = SiteNodeGeometryCleaner(self.subsite)
cleaner_subsite.process_level_curves()
cleaner_mainsite.import_cleaned_geometries_from(cleaner_subsite)
self.assertTrue(cleaner_mainsite.geom['{Level curve B}'].equals(
self.cleaned_level_B_shape))
def test_add_and_clean_level_curves(self):
cleaner = SiteNodeGeometryCleaner(self.mainsite)
cleaner.process_level_curves()
self.assertTrue(cleaner.geom['{Level curve A}'].equals(
self.cleaned_level_A_shape))
# Not directly in sitenode
with self.assertRaises(KeyError):
cleaner['{Level curve B}']
# Already clean
self.assertTrue(cleaner.geom['{Water body}'].equals(
self.waterbody.shape))