def test_two_region_rectangular_board(self):
board = Board([" B "])
territories = board.territories()
self.assertSetEqual(territories[BLACK], {(0, 0), (2, 0)})
self.assertSetEqual(territories[WHITE], set())
self.assertSetEqual(territories[NONE], set())
def test_black_corner_territory_on_5x5_board(self):
board = Board([" B ", " B B ", "B W B", " W W ", " W "])
stone, territory = board.territory(x=0, y=1)
self.assertEqual(stone, BLACK)
self.assertSetEqual(territory, {(0, 0), (0, 1), (1, 0)})
def test_one_territory_is_the_whole_board(self):
board = Board([" "])
territories = board.territories()
self.assertSetEqual(territories[BLACK], set())
self.assertSetEqual(territories[WHITE], set())
self.assertSetEqual(territories[NONE], {(0, 0)})
def test_two_territory_rectangular_board(self):
board = Board([" BW ", " BW "])
territories = board.territories()
self.assertSetEqual(territories[BLACK], {(0, 0), (0, 1)})
self.assertSetEqual(territories[WHITE], {(3, 0), (3, 1)})
self.assertSetEqual(territories[NONE], set())
def test_invalid_because_y_is_too_high_for_5x5_board(self):
board = Board([" B ", " B B ", "B W B", " W W ", " W "])
with self.assertRaisesWithMessage(ValueError):
board.territory(x=1, y=5)
def test_a_stone_and_not_a_territory_on_5x5_board(self):
board = Board([" B ", " B B ", "B W B", " W W ", " W "])
stone, territory = board.territory(x=1, y=1)
self.assertEqual(stone, NONE)
self.assertSetEqual(territory, set())
def test_open_corner_territory_on_5x5_board(self):
board = Board([" B ", " B B ", "B W B", " W W ", " W "])
stone, territory = board.territory(x=1, y=4)
self.assertEqual(stone, NONE)
self.assertSetEqual(territory, {(0, 3), (0, 4), (1, 4)})
def test_white_center_territory_on_5x5_board(self):
board = Board([" B ", " B B ", "B W B", " W W ", " W "])
stone, territory = board.territory(x=2, y=3)
self.assertEqual(stone, WHITE)
self.assertSetEqual(territory, {(2, 3)})
def test_invalid_because_y_is_too_high_for_5x5_board(self):
board = Board([" B ", " B B ", "B W B", " W W ", " W "])
with self.assertRaises(ValueError) as err:
board.territory(x=1, y=5)
self.assertEqual(type(err.exception), ValueError)
self.assertEqual(err.exception.args[0], "Invalid coordinate")
