def test_to_parquet_with_lti(): es = load_mock_customer(return_entityset=True, random_seed=0) path = os.path.join(CACHE, 'es') os.makedirs(path) es.to_parquet(path) new_es = deserialize.read_entityset(path) assert es.__eq__(new_es, deep=True) shutil.rmtree(path)
def test_mock_customer(): n_customers = 4 n_products = 3 n_sessions = 30 n_transactions = 400 es = load_mock_customer(n_customers=n_customers, n_products=n_products, n_sessions=n_sessions, n_transactions=n_transactions, random_seed=0, return_entityset=True) entity_names = [entity.id for entity in es.entities] expected_names = ['transactions', 'products', 'sessions', 'customers'] assert set(expected_names) == set(entity_names) assert len(es['customers'].df) == 4 assert len(es['products'].df) == 3 assert len(es['sessions'].df) == 30 assert len(es['transactions'].df) == 400
def test_mock_customer(): n_customers = 4 n_products = 3 n_sessions = 30 n_transactions = 400 es = load_mock_customer(n_customers=n_customers, n_products=n_products, n_sessions=n_sessions, n_transactions=n_transactions, random_seed=0, return_entityset=True) df_names = [df.ww.name for df in es.dataframes] expected_names = ['transactions', 'products', 'sessions', 'customers'] assert set(expected_names) == set(df_names) assert len(es['customers']) == 4 assert len(es['products']) == 3 assert len(es['sessions']) == 30 assert len(es['transactions']) == 400
def test_mock_customer(): n_customers = 4 n_products = 3 n_sessions = 30 n_transactions = 400 es = load_mock_customer( n_customers=n_customers, n_products=n_products, n_sessions=n_sessions, n_transactions=n_transactions, random_seed=0, return_entityset=True, ) assert isinstance(es, EntitySet) df_names = [df.ww.name for df in es.dataframes] expected_names = ["transactions", "products", "sessions", "customers"] assert set(expected_names) == set(df_names) assert len(es["customers"]) == 4 assert len(es["products"]) == 3 assert len(es["sessions"]) == 30 assert len(es["transactions"]) == 400
def test_mock_customer(): es = load_mock_customer(return_entityset=True) fm, fl = dfs(entityset=es, target_entity="customers", max_depth=3) for feature in fl: assert feature.get_name() in fm.columns
def test_to_parquet_with_lti(tmpdir): es = load_mock_customer(return_entityset=True, random_seed=0) es.to_parquet(str(tmpdir)) new_es = deserialize.read_entityset(str(tmpdir)) assert es.__eq__(new_es, deep=True)
def test_to_parquet_with_lti(path_management): es = load_mock_customer(return_entityset=True, random_seed=0) es.to_parquet(path_management) new_es = deserialize.read_entityset(path_management) assert es.__eq__(new_es, deep=True)