def test_ok(self):
state = GameState()
state.add_robot((9, 9), 0)
self.assertTrue(state.is_valid_action((9, 9), ['guard']))
self.assertTrue(state.is_valid_action((9, 9), ['suicide']))
self.assertTrue(state.is_valid_action((9, 9), ['attack', (9, 10)]))
self.assertTrue(state.is_valid_action((9, 9), ['move', (9, 10)]))
def test_repr_too_long(self):
state = GameState()
state.add_robot((9, 9), 0)
class long_action(list):
def __repr__(self):
return 'leethaxxor' * 100
action = long_action(['guard'])
self.assertFalse(state.is_valid_action((9, 9), action))
def test_attack_obstacle(self):
state = GameState()
state.add_robot((2, 5), 0)
self.assertFalse(state.is_valid_action((2, 5), ['attack', (2, 4)]))
def test_attack_too_far(self):
state = GameState()
state.add_robot((9, 9), 0)
self.assertFalse(state.is_valid_action((9, 9), ['attack', (9, 11)]))
def test_move_to_self(self):
state = GameState()
state.add_robot((9, 9), 0)
self.assertFalse(state.is_valid_action((9, 9), ['move', (9, 9)]))
def test_additional_info(self):
state = GameState()
state.add_robot((9, 9), 0)
self.assertTrue(state.is_valid_action((9, 9),
['guard', 'additional info']))
def test_wrong_command(self):
state = GameState()
state.add_robot((9, 9), 0)
self.assertFalse(state.is_valid_action((9, 9), ['exterminate']))
def test_strange(self):
state = GameState()
state.add_robot((9, 9), 0)
self.assertFalse(state.is_valid_action((9, 9),
"ALL YOUR BASE ARE BELONG TO US"))
def test_none(self):
state = GameState()
state.add_robot((9, 9), 0)
self.assertFalse(state.is_valid_action((9, 9), None))
