-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Expand file tree
/
Copy pathtest_feature.py
More file actions
32 lines (24 loc) · 1.06 KB
/
test_feature.py
File metadata and controls
32 lines (24 loc) · 1.06 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
from feast.field import Feature, Field
from feast.types import Float32
from feast.value_type import ValueType
def test_feature_serialization_with_description():
expected_description = "Average daily trips"
feature = Feature(
name="avg_daily_trips", dtype=ValueType.FLOAT, description=expected_description
)
serialized_feature = feature.to_proto()
assert serialized_feature.description == expected_description
def test_field_serialization_with_description():
expected_description = "Average daily trips"
field = Field(
name="avg_daily_trips", dtype=Float32, description=expected_description
)
feature = Feature(
name="avg_daily_trips", dtype=ValueType.FLOAT, description=expected_description
)
serialized_field = field.to_proto()
field_from_feature = Field.from_feature(feature)
assert serialized_field.description == expected_description
assert field_from_feature.description == expected_description
field = Field.from_proto(serialized_field)
assert field.description == expected_description