def test_neighboring_points_simple(self):
game = Game(6, 6)
game.field[3][2] = Cell.RED
best_path_and_squre = game.check_neighboring_points(
3, 2, 3, 2, [((3, 2))], [], 0, time.time())
self.assertEqual(best_path_and_squre, (([], 0)))
game.field[3][4] = Cell.RED
game.field[2][3] = Cell.RED
game.field[3][3] = Cell.BLUE
game.field[4][3] = Cell.RED
best_path_and_squre = game.check_neighboring_points(
4, 3, 4, 3, [((4, 3))], [], 0, time.time())
self.assertEqual(best_path_and_squre, (([(4, 3), (3, 2), (2, 3),
(3, 4)], 2.0)))
def test_neighboring_points_vertical_border(self):
game = Game(5, 5)
game.field[0][2] = Cell.BLUE
game.field[4][2] = Cell.BLUE
game.field[0][1] = Cell.RED
game.field[1][2] = Cell.RED
game.field[3][2] = Cell.RED
game.field[0][3] = Cell.RED
game.field[4][3] = Cell.RED
game.field[4][1] = Cell.RED
best_path_and_squre = game.check_neighboring_points(
4, 1, 4, 1, [((4, 1))], [], 0, time.time())
self.assertEqual(best_path_and_squre,
(([(4, 1), (3, 2), (4, 3), (5, 3), (6, 2),
(5, 1)], 4.0)))
best_path_and_squre = game.check_neighboring_points(
0, 3, 0, 3, [((0, 3))], [], 0, time.time())
self.assertEqual(best_path_and_squre, (([(0, 3), (-1, 3), (-2, 2),
(-1, 1), (0, 1),
(1, 2)], 4.0)))
def test_neighboring_points_horiz_border(self):
game = Game(5, 5)
game.field[1][0] = Cell.BLUE
game.field[1][4] = Cell.BLUE
game.field[1][1] = Cell.RED
game.field[0][4] = Cell.RED
game.field[1][3] = Cell.RED
game.field[0][0] = Cell.RED
game.field[2][0] = Cell.RED
game.field[2][4] = Cell.RED
best_path_and_squre = game.check_neighboring_points(
2, 4, 2, 4, [((2, 4))], [], 0, time.time())
self.assertEqual(best_path_and_squre,
(([(2, 4), (1, 3), (0, 4), (0, 5), (1, 6),
(2, 5)], 4.0)))
best_path_and_squre = game.check_neighboring_points(
0, 0, 0, 0, [((0, 0))], [], 0, time.time())
self.assertEqual(best_path_and_squre, (([(0, 0), (1, 1),
(2, 0), (2, -1), (1, -2),
(0, -1)], 4.0)))