class TestToyRobotMovement(unittest.TestCase):
def setUp(self):
self.toy_robot = ToyRobot()
self.toy_robot.set_position(0, 0, 'EAST')
def test_move_and_rotate(self):
self.toy_robot.move()
self.toy_robot.move()
self.toy_robot.rotate_left()
self.toy_robot.move()
self.toy_robot.move()
self.assertEqual(self.toy_robot.position.x, 2)
self.assertEqual(self.toy_robot.position.y, 2)
self.assertEqual(self.toy_robot.position.direction, 'NORTH')
class Simulation():
"""Simulation
Represents one simulation in which for a given text one toy_robot is created and
does the commands specified.
After executing all the commands the results of each REPORT command are outputed to stdout.
"""
def __init__(self, command_text):
self.commands = get_commands_from_input(command_text)
self.output_texts = []
self.toy_robot = ToyRobot()
self.movement_dict = {
'LEFT': self.toy_robot.rotate_left,
'RIGHT': self.toy_robot.rotate_right,
'MOVE': self.toy_robot.move
}
self.execute_commands()
self.output_results()
def execute_commands(self):
"""execute_commands
For each command in self.commands call the appropriate ToyRobot action
"""
for command in self.commands:
if command in self.movement_dict:
self.movement_dict[command]()
elif command == 'REPORT':
report = self.toy_robot.report()
if report:
self.output_texts.append(report)
elif command.startswith('PLACE'):
try:
x, y, direction = parse_place_command(command)
self.toy_robot.set_position(x, y, direction)
except InputException:
'''Exception
Silently ignore it to continue robot movement
'''
continue
def output_results(self):
"""output_results
Output all the texts available in self.output_texts which were filled
in execute_commands
"""
for output_text in self.output_texts:
print(output_text)
class TestToyRobotPosition(unittest.TestCase):
def setUp(self):
self.toy_robot = ToyRobot()
self.toy_robot.set_position(0, 0, 'EAST')
def test_rotate_left(self):
self.toy_robot.rotate_left()
self.assertEqual(self.toy_robot.position.direction, 'NORTH')
def test_rotate_right(self):
self.toy_robot.rotate_right()
self.assertEqual(self.toy_robot.position.direction, 'SOUTH')
def test_set_position(self):
self.assertEqual(self.toy_robot.position.x, 0)
self.assertEqual(self.toy_robot.position.y, 0)
self.assertEqual(self.toy_robot.position.direction, 'EAST')