def get_expected_last_state(self):
     if self._direction == Direction.south_east():
         return self.south_east_state
     elif self._direction == Direction.south_west():
         return self.south_west_state
     elif self._direction == Direction.north_east():
         return self.north_east_state
     elif self._direction == Direction.north_west():
         return self.north_west_state
 def get_expected_result(self):
     if self._direction == Direction.south_east():
         return self.south_east_output
     elif self._direction == Direction.south_west():
         return self.south_west_output
     elif self._direction == Direction.north_east():
         return self.north_east_output
     elif self._direction == Direction.north_west():
         return self.north_west_output