Skip to content

Commit

Permalink
Add regression tests for GeographicPartition default_updaters missing…
Browse files Browse the repository at this point in the history
… bug
  • Loading branch information
InnovativeInventor committed Apr 2, 2022
1 parent ee58bd6 commit e78a155
Showing 1 changed file with 24 additions and 1 deletion.
25 changes: 24 additions & 1 deletion tests/partition/test_partition.py
Expand Up @@ -52,7 +52,8 @@ def example_geographic_partition():
graph.nodes[node]["area"] = 1
for edge in graph.edges:
graph.edges[edge]["shared_perim"] = 1
return GeographicPartition(graph, assignment, None, None, None)
partition = GeographicPartition(graph, assignment, None, None, None)
return partition


def test_geographic_partition_can_be_instantiated(example_geographic_partition):
Expand Down Expand Up @@ -153,3 +154,25 @@ def test_partition_has_default_updaters(example_partition):

def test_partition_has_keys(example_partition):
assert "cut_edges" in set(example_partition.keys())


def test_geographic_partition_has_keys(example_geographic_partition):
keys = set(example_geographic_partition.updaters.keys())

assert "perimeter" in keys
assert "exterior_boundaries" in keys
assert "interior_boundaries" in keys
assert "boundary_nodes" in keys
assert "cut_edges" in keys
assert "area" in keys
assert "cut_edges_by_part" in keys


def test_partition_has_default_updaters(example_geographic_partition):
assert hasattr(example_geographic_partition, "perimeter")
assert hasattr(example_geographic_partition, "exterior_boundaries")
assert hasattr(example_geographic_partition, "interior_boundaries")
assert hasattr(example_geographic_partition, "boundary_nodes")
assert hasattr(example_geographic_partition, "cut_edges")
assert hasattr(example_geographic_partition, "area")
assert hasattr(example_geographic_partition, "cut_edges_by_part")

0 comments on commit e78a155

Please sign in to comment.