def test_action_reach_treasure(self):
game_map = Map(width=1, height=2)
fake_map = [
[TreasureField()],
[BotField(Orientation.NORTH)],
]
setattr(game_map, '_{}__map'.format(game_map.__class__.__name__), fake_map)
game = Game(game_map)
with self.assertRaises(GameFinished):
game.action('bot_id', Action.STEP)
def test_action_step_on_bot(self):
game_map = Map(width=1, height=2)
fake_map = [
[BotField(Orientation.SOUTH)],
[BotField(Orientation.NORTH)],
]
setattr(game_map, '_{}__map'.format(game_map.__class__.__name__), fake_map)
game = Game(game_map)
with self.assertRaises(MovementError):
game.action('bot_id', Action.STEP)
def test_action_reach_treasure(self):
game_map = Map(width=1, height=2)
fake_map = [
[TreasureField()],
[BotField(Orientation.NORTH)],
]
setattr(game_map, '_{}__map'.format(game_map.__class__.__name__), fake_map)
game = Game(game_map)
with self.assertRaises(GameFinished):
game.action('bot_id', Action.STEP)
def test_action_step_on_block(self):
game_map = Map(width=1, height=2)
fake_map = [
[BlockField()],
[BotField(Orientation.NORTH)],
]
setattr(game_map, '_{}__map'.format(game_map.__class__.__name__), fake_map)
game = Game(game_map)
with self.assertRaises(MovementError):
game.action('bot_id', Action.STEP)
def test_export(self):
game_map = MapFactory().create(DefaultConfiguration())
game = Game(game_map)
self.assertCountEqual(
game.export(0),
dict(
map=game_map.export(),
bots=game._export_bots(0),
map_width=game_map.width,
map_height=game_map.height,
map_resolutions=(game_map.width, game_map.height)
)
)
def test_laser_beam_bot_fight(self):
class Conf(BaseConfiguration, RandomFieldPlacerMixin):
map_width = 1
map_height = 3
bots = 1
treasures = 0
blocks = 0
battery_game = True
laser_game = True
conf = Conf()
game_map = Map(width=3, height=1)
fake_map = [
[LaserBatteryBotField(Orientation.SOUTH)],
[EmptyField()],
[LaserBatteryBotField(Orientation.NORTH)]
]
setattr(game_map, '_{}__map'.format(game_map.__class__.__name__), fake_map)
game = Game(game_map, configuration=conf)
bot_id = 1
game._bots_positions = {
bot_id: (0, 0)
}
game.action(bot_id, Action.LASER_BEAM)
self.assertIsInstance(
game.map[0, 0],
LaserBatteryBotField
)
self.assertIsInstance(
game.map[0, 1],
EmptyField
)
self.assertIsInstance(
game.map[0, 2],
LaserBatteryBotField
)
defender = game.map[0, 2]
assert isinstance(defender, LaserBatteryBotField)
self.assertEqual(
LaserBatteryBotField.DEFAULT_BATTERY_LEVEL - LaserBatteryBotField.DEFAULT_LASER_DAMAGE,
defender.actual_battery_level
)
attacker = game.map[0, 0]
assert isinstance(attacker, LaserBatteryBotField)
self.assertEqual(
LaserBatteryBotField.DEFAULT_BATTERY_LEVEL - LaserBatteryBotField.DEFAULT_LASER_BATTERY_COST,
attacker.actual_battery_level
)
def test_get_bot_position(self):
class Conf(BaseConfiguration, RandomFieldPlacerMixin):
map_width = 1
map_height = 1
bots = 1
treasures = 0
blocks = 0
conf = Conf()
game = Game(MapFactory.create(conf), configuration=conf)
bot_id = 1
game.add_bot(bot_id)
self.assertEqual(
game.get_bot_position(bot_id),
(0, 0)
)
def get(self, bot_id):
assert isinstance(bot_id, int)
game = self.games.get(bot_id, None)
if game:
return game
# try to find game with free bot
game = self.open_game
if game:
game.add_bot(bot_id)
self.games.update({bot_id: game})
return game
game = Game(MapFactory.create(self._conf_provider.actual))
game.add_bot(bot_id)
self.games.update({bot_id: game})
return game
def test_game_time_flags(self):
dt = datetime.now()
game = Game(Map(1, 1))
self.assertAlmostEqual(
game.created_at.timestamp(),
dt.timestamp(),
places=0,
msg='created game at patched datetime'
)
def get(self, bot_id):
assert isinstance(bot_id, int)
game = self.games.get(bot_id, None)
if game:
return game
# try to find game with free bot
game = self.open_game
if game:
game.add_bot(bot_id)
self.games.update({
bot_id: game
})
return game
game = Game(MapFactory.create(self._conf_provider.actual))
game.add_bot(bot_id)
self.games.update({bot_id: game})
return game
def test_export_bots(self):
class Conf(DefaultConfiguration):
map_width = 2
map_height = 1
bots = 2
treasures = 0
blocks = 0
game = Game(MapFactory().create(Conf()))
my_bot_id = 0
bots_export = game._export_bots(my_bot_id)
my_bot_on_first_field = game.get_bot_position(my_bot_id) == (0, 0)
self.assertListEqual(
bots_export,
[
dict(x=0, y=0, position=(0, 0), orientation=Orientation.NORTH, your_bot=not my_bot_on_first_field),
dict(x=1, y=0, position=(1, 0), orientation=Orientation.NORTH, your_bot=my_bot_on_first_field)
]
)
def test_export(self):
game_map = MapFactory.create(DefaultConfiguration())
game = Game(game_map)
bot_id = 0
game.add_bot(bot_id)
self.assertCountEqual(
game.export(bot_id),
dict(
map=game_map.export(),
game_info=dict(
map_resolutions=dict(
width=game_map.width,
height=game_map.height
),
rounded_game=DefaultConfiguration.rounded_game,
battery_game=DefaultConfiguration.battery_game,
laser_game=DefaultConfiguration.laser_game
),
)
)
def test_invalid_action_in_laser_game(self):
class Conf(BaseConfiguration, RandomFieldPlacerMixin):
map_width = 1
map_height = 1
bots = 1
treasures = 0
blocks = 0
battery_game = True
laser_game = False
conf = Conf()
game = Game(MapFactory.create(conf), configuration=conf)
bot_id = 1
game.add_bot(bot_id)
with self.assertRaises(ActionError):
game.action(bot_id, Action.LASER_BEAM)
def test_draining_battery_in_battery_game(self):
class Conf(BaseConfiguration, RandomFieldPlacerMixin):
map_width = 12
map_height = 1
bots = 1
treasures = 0
blocks = 0
battery_game = True
conf = Conf()
game_map = Map(width=12, height=1)
fake_map = [
[LaserBatteryBotField(Orientation.SOUTH)],
]
fake_map.extend([[EmptyField()] for _ in range(11)])
setattr(game_map, '_{}__map'.format(game_map.__class__.__name__), fake_map)
game = Game(game_map, configuration=conf)
bot_id = 1
game.add_bot(bot_id)
with self.assertRaises(CriticalBatteryLevel):
while True:
game.action(bot_id, Action.STEP)
def test_laser_beam_action_block_destroy(self):
class Conf(BaseConfiguration, RandomFieldPlacerMixin):
map_width = 1
map_height = 3
bots = 1
treasures = 0
blocks = 0
battery_game = True
laser_game = True
conf = Conf()
game_map = Map(width=3, height=1)
fake_map = [
[LaserBatteryBotField(Orientation.SOUTH)],
[EmptyField()],
[BlockField()]
]
setattr(game_map, '_{}__map'.format(game_map.__class__.__name__), fake_map)
game = Game(game_map, configuration=conf)
bot_id = 1
game.action(bot_id, Action.LASER_BEAM)
self.assertIsInstance(
game.map[0, 0],
LaserBatteryBotField
)
self.assertIsInstance(
game.map[0, 1],
EmptyField
)
self.assertIsInstance(
game.map[0, 2],
EmptyField
)
with self.assertRaises(MovementError):
while True:
game.action(bot_id, Action.LASER_BEAM)
def test_not_free_bots(self):
game_map = MapFactory.create(CustomConfiguration(map_width=2, map_height=1, bots=0, treasures=0, blocks=0))
game = Game(game_map)
with self.assertRaises(NoFreeBots):
game.action('bot_id', Action.STEP)
self.assertTrue(game.is_filled, 'Game is filled.')
def test_rounded_game(self):
class Conf(BaseConfiguration, RandomFieldPlacerMixin):
map_width = 3
map_height = 1
bots = 3
treasures = 0
blocks = 0
rounded_game = True
conf = Conf()
game = Game(MapFactory.create(conf), configuration=conf)
my_bot_1 = 1
my_bot_2 = 2
my_bot_3 = 3
game.add_bot(my_bot_1)
game.add_bot(my_bot_2)
game.action(my_bot_1, Action.TURN_LEFT)
game.action(my_bot_2, Action.TURN_LEFT)
with self.assertRaises(BotNotOnTurn):
game.action(my_bot_2, Action.TURN_LEFT)
game.action(my_bot_1, Action.TURN_LEFT)
with self.assertRaises(BotNotOnTurn):
game.action(my_bot_1, Action.TURN_LEFT)
game.add_bot(my_bot_3)
with self.assertRaises(BotNotOnTurn):
game.action(my_bot_3, Action.TURN_LEFT)
game.action(my_bot_2, Action.TURN_LEFT)
game.action(my_bot_1, Action.TURN_LEFT)
game.action(my_bot_3, Action.TURN_LEFT)
def test_action_simple_movements(self):
game_map = Map(width=1, height=1)
fake_map = [
[BotField(Orientation.EAST), EmptyField()],
[EmptyField(), EmptyField()],
]
setattr(game_map, '_{}__map'.format(game_map.__class__.__name__), fake_map)
game = Game(game_map)
game._empty_bots_positions = game._map.get_field_occurrences(BotField)
self.assertFalse(game.is_filled, 'Game has free bot field.')
fake_bot_id = 'fake_bot_id'
game.action(fake_bot_id, Action.TURN_RIGHT)
self.assertEqual(
game._map[game._bots_positions[fake_bot_id]].orientation,
Orientation.SOUTH
)
game.action(fake_bot_id, Action.TURN_LEFT)
self.assertEqual(
game.map[game._bots_positions[fake_bot_id]].orientation,
Orientation.EAST
)
game.action(fake_bot_id, Action.STEP)
self.assertEqual(
game._bots_positions[fake_bot_id],
(1, 0)
)
with self.assertRaises(MovementError):
game.action(fake_bot_id, Action.STEP)
self.assertEqual(
game._bots_positions[fake_bot_id],
(1, 0)
)
game.action(fake_bot_id, Action.TURN_RIGHT)
game.action(fake_bot_id, Action.STEP)
with self.assertRaises(MovementError):
game.action(fake_bot_id, Action.STEP)
def test_not_free_bots(self):
game_map = MapFactory().create(CustomConfiguration(map_width=2, map_height=1, bots=0, treasures=0, blocks=0))
game = Game(game_map)
with self.assertRaises(NoFreeBots):
game.action('bot_id', Action.STEP)
self.assertTrue(game.is_filled, 'Game is filled.')
def test_rounded_game(self):
class Conf(DefaultConfiguration):
map_width = 3
map_height = 1
bots = 3
treasures = 0
blocks = 0
rounded_game = True
conf = Conf()
game = Game(MapFactory().create(conf), configuration=conf)
my_bot_1 = 1
my_bot_2 = 2
my_bot_3 = 3
game.add_bot(my_bot_1)
game.add_bot(my_bot_2)
game.action(my_bot_1, Action.TURN_LEFT)
game.action(my_bot_2, Action.TURN_LEFT)
with self.assertRaises(BotNotOnTurn):
game.action(my_bot_2, Action.TURN_LEFT)
game.action(my_bot_1, Action.TURN_LEFT)
with self.assertRaises(BotNotOnTurn):
game.action(my_bot_1, Action.TURN_LEFT)
game.add_bot(my_bot_3)
with self.assertRaises(BotNotOnTurn):
game.action(my_bot_3, Action.TURN_LEFT)
game.action(my_bot_2, Action.TURN_LEFT)
game.action(my_bot_1, Action.TURN_LEFT)
game.action(my_bot_3, Action.TURN_LEFT)
def test_charging_battery_in_battery_game(self):
class Conf(BaseConfiguration, RandomFieldPlacerMixin):
map_width = 1
map_height = 3
bots = 1
treasures = 0
blocks = 0
battery_game = True
conf = Conf()
game_map = Map(width=3, height=1)
fake_map = [
[LaserBatteryBotField(Orientation.EAST)],
[EmptyField()],
[EmptyField()]
]
setattr(game_map, '_{}__map'.format(game_map.__class__.__name__), fake_map)
game = Game(game_map, configuration=conf)
my_bot_id = 1
game.add_bot(my_bot_id)
bot_field = game.map[game._bots_positions[my_bot_id]]
game.action(my_bot_id, Action.TURN_RIGHT)
assert isinstance(bot_field, LaserBatteryBotField)
self.assertEqual(LaserBatteryBotField.DEFAULT_BATTERY_LEVEL,
bot_field.actual_battery_level,
'Default battery level')
game.action(my_bot_id, Action.STEP)
self.assertEqual(LaserBatteryBotField.DEFAULT_BATTERY_LEVEL - 1,
bot_field.actual_battery_level,
'Drained battery level')
game.action(my_bot_id, Action.WAIT)
self.assertEqual(LaserBatteryBotField.DEFAULT_BATTERY_LEVEL,
bot_field.actual_battery_level,
'Charged battery level to default.')
game.action(my_bot_id, Action.WAIT)
self.assertEqual(LaserBatteryBotField.DEFAULT_BATTERY_LEVEL + 1,
bot_field.actual_battery_level,
'Charged battery level to higher level than default.')
def test_action_simple_movements(self):
game_map = Map(width=1, height=1)
fake_map = [
[BotField(Orientation.EAST), EmptyField()],
[EmptyField(), EmptyField()],
]
setattr(game_map, '_{}__map'.format(game_map.__class__.__name__), fake_map)
game = Game(game_map)
game._empty_bots_positions = game._map.get_field_occurrences(BotField)
self.assertFalse(game.is_filled, 'Game has free bot field.')
fake_bot_id = 'fake_bot_id'
game.action(fake_bot_id, Action.TURN_RIGHT)
self.assertEqual(
game._map[game._bots_positions[fake_bot_id]].orientation,
Orientation.SOUTH
)
game.action(fake_bot_id, Action.TURN_LEFT)
self.assertEqual(
game.map[game._bots_positions[fake_bot_id]].orientation,
Orientation.EAST
)
game.action(fake_bot_id, Action.STEP)
self.assertEqual(
game._bots_positions[fake_bot_id],
(1, 0)
)
with self.assertRaises(MovementError):
game.action(fake_bot_id, Action.STEP)
self.assertEqual(
game._bots_positions[fake_bot_id],
(1, 0)
)
game.action(fake_bot_id, Action.TURN_RIGHT)
game.action(fake_bot_id, Action.STEP)
with self.assertRaises(MovementError):
game.action(fake_bot_id, Action.STEP)