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)