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