Skip to content

Commit

Permalink
Fix hydrofab subset test for abstraction change.
Browse files Browse the repository at this point in the history
Fixing setup for TestSimpleHydrofabricSubset that used the
MappedGraphHydrofabric class after it was made abstract, replacing its
use with its GeoJsonHydrofabric subtype.
  • Loading branch information
robertbartel committed Jan 11, 2023
1 parent de86117 commit 9f4757c
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from pathlib import Path
from typing import Optional, Union
from ..modeldata.subset import SimpleHydrofabricSubset, SubsetHandler
from ..modeldata.hydrofabric import GeoJsonHydrofabricReader, MappedGraphHydrofabric
from ..modeldata.hydrofabric import GeoJsonHydrofabricReader, GeoJsonHydrofabric


class TestSimpleHydrofabricSubset(unittest.TestCase):
Expand Down Expand Up @@ -53,7 +53,7 @@ def setUp(self) -> None:

geojson_reader = GeoJsonHydrofabricReader(catchment_data=catchment_geojson, nexus_data=nexus_geojson,
cross_walk=crosswalk_json)
self.hydrofabric = MappedGraphHydrofabric(geojson_reader.hydrofabric_graph, geojson_reader.roots, geojson_reader)
self.hydrofabric = GeoJsonHydrofabric(geojson_reader)
self.subset_handler = SubsetHandler(self.hydrofabric)

self.subset_examples = list()
Expand Down

0 comments on commit 9f4757c

Please sign in to comment.