def test_pull_no_car(): road = Road([ [ [None, None, None], [None, None, None], ], [ [Car(Directions.BOTTOM, TurnDirection.LEFT), Car(Directions.BOTTOM, TurnDirection.LEFT), None], [Car(Directions.BOTTOM, TurnDirection.LEFT), Car(Directions.BOTTOM, TurnDirection.LEFT), None], ] ]) assert road.pull_car(0) is None assert road.pull_car(1) is None
def test_pull_car(): road = Road([ [ [None, None, None], [None, None, None], ], [ [None, None, Car(Directions.BOTTOM, TurnDirection.LEFT)], [None, None, Car(Directions.TOP, TurnDirection.LEFT)], ] ]) car1 = road.pull_car(1) car0 = road.pull_car(0) assert car0 is not None assert isinstance(car0, Car) assert car0.source == Directions.BOTTOM assert car1 is not None assert isinstance(car1, Car) assert car1.source == Directions.TOP