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 18, 2023
1 parent 2a4a922 commit 300c1bf
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 300c1bf

Please sign in to comment.