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 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_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 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 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(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_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_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_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)