def test_to_pickle_id_none():
path = os.path.join(CACHE, 'es')
os.makedirs(path)
es = EntitySet()
es.to_pickle(path)
new_es = deserialize.read_entityset(path)
assert es.__eq__(new_es, deep=True)
shutil.rmtree(path)
def test_all_ww_logical_types():
logical_types = list_logical_types()['type_string'].to_list()
dataframe = pd.DataFrame(columns=logical_types)
es = EntitySet()
ltype_dict = {ltype: ltype for ltype in logical_types}
ltype_dict['ordinal'] = Ordinal(order=[])
es.add_dataframe(dataframe=dataframe,
dataframe_name='all_types',
index='integer',
logical_types=ltype_dict)
description = serialize.entityset_to_description(es)
_es = deserialize.description_to_entityset(description)
assert es.__eq__(_es, deep=True)
def test_with_custom_ww_logical_type():
class CustomLogicalType(LogicalType):
pass
ww_type_system.add_type(CustomLogicalType)
columns = ['integer', 'natural_language', 'custom_logical_type']
dataframe = pd.DataFrame(columns=columns)
es = EntitySet()
ltype_dict = {
'integer': 'integer',
'natural_language': 'natural_language',
'custom_logical_type': CustomLogicalType,
}
es.add_dataframe(dataframe=dataframe,
dataframe_name='custom_type',
index='integer',
logical_types=ltype_dict)
description = serialize.entityset_to_description(es)
_es = deserialize.description_to_entityset(description)
assert isinstance(
_es['custom_type'].ww.logical_types['custom_logical_type'],
CustomLogicalType)
assert es.__eq__(_es, deep=True)
def test_to_pickle_id_none(tmpdir):
es = EntitySet()
es.to_pickle(str(tmpdir))
new_es = deserialize.read_entityset(str(tmpdir))
assert es.__eq__(new_es, deep=True)
def test_to_pickle_id_none(path_management):
es = EntitySet()
es.to_pickle(path_management)
new_es = deserialize.read_entityset(path_management)
assert es.__eq__(new_es, deep=True)