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)
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
