def test_apply_action_collide_with_ship(self): ship1 = Ship('1', Location(150, 100), 0, 8, 6) ship2 = Ship('2', Location(155, 100), 0, 8, 6) environment = Environment(300, 200, [ship1, ship2]) action = Action(ship1.uid, 0, True) with self.assertRaises(Exception): environment.apply_action(action)
def test_apply_action(self): ship = Ship('uid', Location(150, 100), 90, 8, 6) environment = Environment(300, 200, [ship]) action = Action(ship.uid, 0, True) environment.apply_action(action) self.assertEqual(Location(150, 94), ship.location) # moved up 6 (the ship's speed)
def test_apply_action_collide_with_right_wall(self): ship = Ship('uid', Location(295, 100), 0, 8, 6) environment = Environment(300, 200, [ship]) action = Action(ship.uid, 0, True) with self.assertRaises(Exception): environment.apply_action(action)