def test_one_way_flight_booking(data):
driver = InitiateDriver.start_browser()
flights = flight_booking.FlightBooking(driver)
flights.book_one_way_ticket(data[0], data[1])
InitiateDriver.close_browser()
def test_booking_bus():
driver = InitiateDriver.start_browser()
hotel = bus_booking.BusBooking(driver)
hotel.book_bus("Mysore", "ooty")
InitiateDriver.close_browser()