def test_can_be_created_from_a_districtr_file(graph, districtr_plan_file):
for node in graph:
graph.nodes[node]["area_num_1"] = node
partition = Partition.from_districtr_file(graph, districtr_plan_file)
assert partition.assignment.to_dict() == {
0: 1,
1: 1,
2: 1,
3: 2,
4: 2,
5: 2,
6: 3,
7: 3,
8: 3,
}
def test_from_districtr_plan_raises_if_id_column_missing(
graph, districtr_plan_file):
with pytest.raises(TypeError):
Partition.from_districtr_file(graph, districtr_plan_file)