def test_create_feature(self):
col = pd.Series([1] * 3, dtype='int32', name="test")
expected = Feature(name="test",
entity="test",
owner="person",
value_type=ValueType.INT32)
actual = _create_feature(col, "test", "person", None, None)
assert actual.id == expected.id
assert actual.value_type == expected.value_type
assert actual.owner == expected.owner
def test_create_feature_with_stores(self):
col = pd.Series([1] * 3, dtype='int32', name="test")
expected = Feature(name="test",
entity="test",
owner="person",
value_type=ValueType.INT32,
serving_store=Datastore(id="SERVING"),
warehouse_store=Datastore(id="WAREHOUSE"))
actual = _create_feature(col, "test", "person",
Datastore(id="SERVING"),
Datastore(id="WAREHOUSE"))
assert actual.id == expected.id
assert actual.value_type == expected.value_type
assert actual.owner == expected.owner
assert actual.serving_store == expected.serving_store
assert actual.warehouse_store == expected.warehouse_store