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}
