def test_load_flight():
es = load_flight(month_filter=[1],
categorical_filter={'origin_city': ['Charlotte, NC']},
return_single_table=False, nrows=1000)
entity_names = ['airports', 'flights', 'trip_logs', 'airlines']
realvals = [(11, 3), (13, 9), (103, 21), (1, 1)]
for i, name in enumerate(entity_names):
assert es[name].shape == realvals[i]
def test_load_flight():
es = load_flight(month_filter=[1],
categorical_filter={'origin_city': ['Charlotte, NC']},
return_single_table=False, nrows=1000)
entity_names = ['airports', 'flights', 'trip_logs', 'airlines']
realvals = [(11, 3), (13, 9), (103, 22), (1, 1)]
for i, name in enumerate(entity_names):
assert es[name].shape == realvals[i]
def test_load_flight():
demo_path, _, _ = make_flight_pathname(demo=True)
es = load_flight(month_filter=[1],
categorical_filter={'origin_city': ['Charlotte, NC']},
return_single_table=False, nrows=1000)
assert os.path.isfile(demo_path)
entity_names = ['airports', 'flights', 'trip_logs', 'airlines']
realvals = [(11, 3), (13, 9), (103, 22), (1, 1)]
for i, name in enumerate(entity_names):
assert es[name].shape == realvals[i]
def test_load_flight():
es = load_flight(
month_filter=[1],
categorical_filter={"origin_city": ["Charlotte, NC"]},
return_single_table=False,
nrows=1000,
)
assert isinstance(es, EntitySet)
dataframe_names = ["airports", "flights", "trip_logs", "airlines"]
realvals = [(11, 3), (13, 9), (103, 21), (1, 1)]
for i, name in enumerate(dataframe_names):
assert es[name].shape == realvals[i]