def test_find_seat_for_wheelchair_users(
self): # add new test - grow the spectrum of testing
finder = SeatFinder(available_seats={'A1W', 'A6', 'B6', 'C7'})
seats = finder.find_seat(1, wheelchair_count=1)
assert seats == {'A1W'}
def test_prefer_near_the_front(self):
finder = SeatFinder(available_seats={'A6', 'B6', 'C7'})
seats = finder.find_seat(1)
assert seats == {'A6'}
def test_find_seats_fails_when_not_enough_available(self):
finder = SeatFinder(available_seats={'A6', 'B6', 'C7'})
seats = finder.find_seat(5)
assert seats == {}
