def test_variable_descriptions(es):
for entity in es.entities:
for variable in entity.variables:
description = variable.to_data_description()
_variable = deserialize.description_to_variable(description,
entity=entity)
assert variable.__eq__(_variable)
def test_custom_description_variable_description(es):
for entity in es.entities:
for variable in entity.variables:
variable.description = 'a custom description'
description = variable.to_data_description()
_variable = deserialize.description_to_variable(description,
entity=entity)
assert _variable.description == 'a custom description'
def test_unknown_variable_description(es):
description = {
'type': 'some_unknown_type',
'id': 'some_unknown_id',
'properties': {
'name': 'some_unknown_type',
'interesting_values': '{}'
}
}
variable = deserialize.description_to_variable(description,
entity=es.entities[0])
assert (variable.dtype == 'unknown')
def test_all_variable_descriptions():
es = EntitySet()
dataframe = pd.DataFrame(columns=list(serialize.VARIABLE_TYPES))
es.entity_from_dataframe(
'variable_types',
dataframe,
index='index',
time_index='datetime_time_index',
variable_types=serialize.VARIABLE_TYPES,
)
entity = es['variable_types']
for variable in entity.variables:
description = variable.to_data_description()
_variable = deserialize.description_to_variable(description,
entity=entity)
assert variable.__eq__(_variable)
def test_custom_variable_descriptions():
class ItemList(Categorical):
type_string = "item_list"
_default_pandas_dtype = list
es = EntitySet()
variables = {'item_list': ItemList, 'time_index': TimeIndex, 'index': Index}
dataframe = pd.DataFrame(columns=list(variables))
es.entity_from_dataframe(
'custom_variable', dataframe, index='index',
time_index='time_index', variable_types=variables)
entity = es['custom_variable']
for variable in entity.variables:
description = variable.to_data_description()
_variable = deserialize.description_to_variable(description, entity=entity)
assert variable.__eq__(_variable)
