def test_get_next_orientation(self): with self.assertRaises(AssertionError): get_next_orientation(0, Action.STEP) with self.assertRaises(AssertionError): get_next_orientation(Orientation.NORTH, 0) self.assertEqual(get_next_orientation(Orientation.NORTH, Action.TURN_RIGHT), Orientation.EAST) self.assertEqual(get_next_orientation(Orientation.NORTH, Action.TURN_LEFT), Orientation.WEST)
def test_get_next_orientation(self): with self.assertRaises(AssertionError): get_next_orientation(0, Action.STEP) with self.assertRaises(AssertionError): get_next_orientation(Orientation.NORTH, 0) self.assertEqual( get_next_orientation(Orientation.NORTH, Action.TURN_RIGHT), Orientation.EAST) self.assertEqual( get_next_orientation(Orientation.NORTH, Action.TURN_LEFT), Orientation.WEST)
def rotate(self, action): assert isinstance(action, Action) self._orientation = get_next_orientation( self._orientation, action )
def rotate(self, action): assert isinstance(action, Action) self._orientation = get_next_orientation(self._orientation, action)