def test_fill_island(self):
# fmt: off
game_map = create_map_with([[1, 1, E, 1], [1, 1, E, 1], [E, E, E, E],
[E, E, E, E]])
# fmt: on
ps = PlayerState(id=1,
name="dummy",
game_map=game_map,
position=Position(2, 2))
gs = GameState(game_map, [ps])
ps.tail = [
ps.spawn_position,
Position(2, 3),
Position(3, 3),
Position(4, 3),
Position(4, 2)
]
self.assertEqual(len(gs.fill(ps)), 3)
# fmt: off
self.assertEqual(
game_map,
create_map_with([[1, 1, E, 1], [1, 1, E, 1], [E, 1, 1, 1],
[E, E, E, E]]))
# fmt: on
self.assertEqual(ps.score, 3)
def test_fill_planet_to_planet_2(self):
# fmt: off
game_map = create_map_with([[E, E, E, P1, 1], [E, E, E, 1, P1],
[E, E, E, E, E], [E, E, E, E, E],
[E, E, E, E, E]])
# fmt: on
ps = PlayerState(id=1,
name="dummy",
game_map=game_map,
position=Position(4, 1))
gs = GameState(game_map, [ps])
ps.tail = [
Position(4, 1),
Position(3, 1),
Position(2, 1),
Position(2, 2),
Position(2, 3),
Position(2, 4),
Position(3, 4),
Position(4, 4),
Position(5, 4),
Position(5, 3),
Position(5, 2),
]
self.assertEqual(len(gs.fill(ps)), 12)
# fmt: off
self.assertEqual(
game_map,
create_map_with([[E, 1, 1, P1, 1], [E, 1, 1, 1,
P1], [E, 1, 1, 1, 1],
[E, 1, 1, 1, 1], [E, E, E, E, E]]))
# fmt: on
self.assertEqual(ps.score, 12)
def test_fill_with_asteroids(self):
# fmt: off
game_map = create_map_with([[E, E, E, E, E], [E, 1, E, 1, E],
[E, 1, W, 1, E], [E, 1, 1, 1, E],
[E, E, E, E, E]])
# fmt: on
ps = PlayerState(id=1,
name="dummy",
game_map=game_map,
position=Position(2, 2))
ps.tail = [ps.spawn_position, Position(3, 2), Position(4, 2)]
gs = GameState(game_map, [ps])
self.assertEqual(len(gs.fill(ps)), 1)
# fmt: off
self.assertEqual(
game_map,
create_map_with([[E, E, E, E, E], [E, 1, 1, 1, E], [E, 1, W, 1, E],
[E, 1, 1, 1, E], [E, E, E, E, E]]))
# fmt: on
self.assertEqual(ps.score, 1)
