def get_valid_directions_on_grid(self, row: int, col: int) -> List[int]: """ Returns directions in which the agent can move Parameters: --------- row : int col : int Returns: ------- List[int] """ return Grid4Transitions.get_entry_directions(self.rail.get_full_transitions(row, col))
def _assert(transition, expected): actual = Grid4Transitions.get_entry_directions(transition) assert actual == expected, "Found {}, expected {}.".format( actual, expected)