FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

chore: Adding unit test to test feature view dummy entity serialization after apply() by franciscojavierarceo · Pull Request #4553 · feast-dev/feast · GitHub

/ feast Public

chore: Adding unit test to test feature view dummy entity serialization after apply() #4553

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
tokoko merged 2 commits into from
Sep 21, 2024
Conversation 1 Commits 2 Checks 0 Files changed
Merged

chore: Adding unit test to test feature view dummy entity serialization after apply() #4553

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a0fafef
chore: Adding unit test to test feature view dummy entity serializati…
franciscojavierarceo Sep 21, 2024
c853205
updated
franciscojavierarceo Sep 21, 2024
File filter

Filter by extension

Filter by extension .py  (1) All 1 file type selected
Viewed files
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Unified
Split
Hide whitespace
Diff view
Unified
Split
Hide whitespace
47 changes: 46 additions & 1 deletion sdk/python/tests/unit/local_feast_tests/test_local_feature_store.py
Show comments
View file
Open in desktop
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
Show hidden characters
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
from feast.entity import Entity
from feast.feast_object import ALL_RESOURCE_TYPES
from feast.feature_store import FeatureStore
from feast.feature_view import FeatureView
from feast.feature_view import DUMMY_ENTITY_ID, FeatureView
from feast.field import Field
from feast.infra.offline_stores.file_source import FileSource
from feast.infra.online_stores.sqlite import SqliteOnlineStoreConfig
Expand Down Expand Up @@ -342,6 +342,51 @@ def test_apply_entities_and_feature_views(test_feature_store):
test_feature_store.teardown()


@pytest.mark.parametrize(
"test_feature_store",
[lazy_fixture("feature_store_with_local_registry")],
)
def test_apply_dummuy_entity_and_feature_view_columns(test_feature_store):
assert isinstance(test_feature_store, FeatureStore)
# Create Feature Views
batch_source = FileSource(
file_format=ParquetFormat(),
path="file://feast/*",
timestamp_field="ts_col",
created_timestamp_column="timestamp",
)

e1 = Entity(name="fs1_my_entity_1", description="something")

fv = FeatureView(
name="my_feature_view_no_entity",
schema=[
Field(name="fs1_my_feature_1", dtype=Int64),
Field(name="fs1_my_feature_2", dtype=String),
Field(name="fs1_my_feature_3", dtype=Array(String)),
Field(name="fs1_my_feature_4", dtype=Array(Bytes)),
Field(name="fs1_my_entity_2", dtype=Int64),
],
entities=[],
tags={"team": "matchmaking"},
source=batch_source,
ttl=timedelta(minutes=5),
)

# Check that the entity_columns are empty before applying
assert fv.entity_columns == []

# Register Feature View
test_feature_store.apply([fv, e1])
fv_actual = test_feature_store.get_feature_view("my_feature_view_no_entity")

# Note that after the apply() the feature_view serializes the Dummy Entity ID
assert fv.entity_columns[0].name == DUMMY_ENTITY_ID
assert fv_actual.entity_columns[0].name == DUMMY_ENTITY_ID

test_feature_store.teardown()


@pytest.mark.parametrize(
"test_feature_store",
[lazy_fixture("feature_store_with_local_registry")],
Expand Down
Back | FazBrowse Home | New Git URL