diff --git a/tests/test_load_functions.py b/tests/test_load_functions.py index 16e46b17..43bef315 100644 --- a/tests/test_load_functions.py +++ b/tests/test_load_functions.py @@ -74,43 +74,3 @@ def check_column_schema(columns, df, metadata): assert col in df.columns assert col in metadata.keys() assert isinstance(metadata[col], ColumnSchema) - - -# def test_load_flight(): -# airlines_df, airport_df, flights_df = load_flight() -# for col in [ -# "IATA_CODE", -# "AIRPORT", -# "CITY", -# "STATE", -# "COUNTRY", -# "LATITUDE", -# "LONGITUDE", -# ]: -# assert col in airport_df.columns -# for col in ["IATA_CODE", "AIRLINE"]: -# assert col in airlines_df.columns -# for col in [ -# "DATE", -# "DAY_OF_WEEK", -# "AIRLINE", -# "FLIGHT_NUMBER", -# "TAIL_NUMBER", -# "ORIGIN_AIRPORT", -# "DESTINATION_AIRPORT", -# "SCHEDULED_DEPARTURE_HOUR", -# "SCHEDULED_TIME", -# "ELAPSED_TIME", -# "DEPARTURE_DELAY", -# "ARRIVAL_DELAY", -# "CANCELLED", -# "CANCELLATION_REASON", -# "AIR_SYSTEM_DELAY", -# "SECURITY_DELAY", -# "AIRLINE_DELAY", -# "LATE_AIRCRAFT_DELAY", -# "WEATHER_DELAY", -# ]: -# assert col in flights_df.columns - -# assert flights_df["DATE"].dtype == "datetime64[ns]"