def test_bad_method_name(self): with self.assertRaises(ValueError): get_branches_by_area(self.grid, ["Oregon"], "purple")
def test_bad_area_name(self): with self.assertRaises(ValueError): get_branches_by_area(self.grid, ["S"], "internal")
def test_bad_method_type(self): with self.assertRaises(TypeError): get_branches_by_area(self.grid, ["Oregon"], ["bridging"])
def test_bad_grid_type(self): with self.assertRaises(TypeError): get_branches_by_area("grid", ["Oregon"], "either")
def test_bad_area_type(self): with self.assertRaises(TypeError): get_branches_by_area(self.grid, "Oregon", "either")
def test_either_E(self): branch_idxs = get_branches_by_area(self.grid, ("Oregon", ), method="either") assert branch_idxs == {101, 102, 103, 104, 105}
def test_either_EW(self): branch_idxs = get_branches_by_area(self.grid, ("Oregon", "Washington"), "either") assert branch_idxs == {101, 102, 103, 104, 105, 106, 107, 108}
def test_bridging_EW(self): branch_idxs = get_branches_by_area(self.grid, ("Washington", "Oregon"), "bridging") assert branch_idxs == {101, 105}
def test_either_W(self): branch_idxs = get_branches_by_area(self.grid, ("Washington", ), method="either") assert branch_idxs == {101, 105, 106, 107, 108}
def test_bridging_E(self): branch_idxs = get_branches_by_area(self.grid, {"Oregon"}, method="bridging") assert branch_idxs == {101, 105}
def test_bridging_W(self): branch_idxs = get_branches_by_area(self.grid, ["Washington"], method="bridging") assert branch_idxs == {101, 105}
def test_internal_EW(self): branch_idxs = get_branches_by_area(self.grid, ("Washington", "Oregon"), "internal") assert branch_idxs == {102, 103, 104, 106, 107, 108}
def test_internal_E(self): branch_idxs = get_branches_by_area(self.grid, ["Oregon"], method="internal") assert branch_idxs == {102, 103, 104}
def test_internal_W(self): branch_idxs = get_branches_by_area(self.grid, {"Washington"}, method="internal") assert branch_idxs == {106, 107, 108}