def test_reefer_positions(): env = RoRoDeck() env.vehicle_data[5][4] = 1 # enable reefer for test env.reset() env.current_Lane = 0 assert env.grid_reefer.T[env.current_Lane][-1] == 1 assert len(env.vehicle_data.T) == len(env.get_possible_actions_of_state()) == 5 env.current_Lane = 4 assert env.grid_reefer.T[env.current_Lane][-1] == 0 assert len(env.get_possible_actions_of_state()) == 4 assert 4 not in env.get_possible_actions_of_state()
def test_shifts_caused(): env = RoRoDeck() env.reset() current_lane = env.current_Lane env.step(0) env.current_Lane = current_lane assert env._get_number_of_shifts(1) == 1 assert env._get_number_of_shifts(0) == 0