def test_get_displacement_two(self): robot = Robot(2, 5, 0) displacement = robot.get_displacement_to_point(3, 3) self.assertEqual(round(displacement, 2), 2.24)