Esempio n. 1
0
def test_to_parquet(entityset):
    dirname = os.path.dirname(integration_data.__file__)
    path = os.path.join(dirname, 'test_entityset.p')
    if os.path.exists(path):
        shutil.rmtree(path)
    entityset.to_parquet(path)
    new_es = ft.read_parquet(path)
    assert entityset.__eq__(new_es, deep=True)
    shutil.rmtree(path)
Esempio n. 2
0
def test_to_parquet(entityset):
    dirname = os.path.dirname(integration_data.__file__)
    path = os.path.join(dirname, 'test_entityset.p')
    if os.path.exists(path):
        shutil.rmtree(path)
    entityset.to_parquet(path)
    new_es = ft.read_parquet(path)
    assert entityset.__eq__(new_es, deep=True)
    shutil.rmtree(path)
Esempio n. 3
0
def test_to_parquet_with_lti():
    entityset = ft.demo.load_mock_customer(return_entityset=True, random_seed=0)
    dirname = os.path.dirname(integration_data.__file__)
    path = os.path.join(dirname, 'test_entityset.p')
    if os.path.exists(path):
        shutil.rmtree(path)
    entityset.to_parquet(path)
    new_es = ft.read_parquet(path)
    assert entityset.__eq__(new_es, deep=True)
    shutil.rmtree(path)
Esempio n. 4
0
def test_to_parquet_with_lti():
    entityset = ft.demo.load_mock_customer(return_entityset=True, random_seed=0)
    dirname = os.path.dirname(integration_data.__file__)
    path = os.path.join(dirname, 'test_entityset.p')
    if os.path.exists(path):
        shutil.rmtree(path)
    entityset.to_parquet(path)
    new_es = ft.read_parquet(path)
    assert entityset.__eq__(new_es, deep=True)
    shutil.rmtree(path)