def choose_arrival_flight(driver, arrival_flight): to_port_dropdown = utils.find_element(driver, By.NAME, "toPort") find_flight_in_dropdown(dropdown_element=to_port_dropdown, expected_flight=arrival_flight)
def choose_departure_flight(driver, departure_flight): from_port_dropdown = utils.find_element(driver, By.NAME, "fromPort") find_flight_in_dropdown(dropdown_element=from_port_dropdown, expected_flight=departure_flight)