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]