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)