WIP

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]"