Function test_diff_odfv

sdk/python/tests/unit/diff/test_registry_diff.py:103–152

(simple_dataset_1)

def test_diff_odfv(simple_dataset_1):
    with prep_file_source(df=simple_dataset_1, timestamp_field="ts_1") as file_source:
        entity = Entity(name="id", join_keys=["id"])
        fv = FeatureView(
            name="fv2",
            entities=[entity],
            source=file_source,
            tags={"when": "before"},
        )

        @on_demand_feature_view(
            sources=[fv],
            schema=[Field(name="first_char", dtype=String)],
        )
        def pre_changed(inputs: pd.DataFrame) -> pd.DataFrame:
            df = pd.DataFrame()
            df["first_char"] = inputs["string_col"].str[:1].astype("string")
            return df

        @on_demand_feature_view(
            sources=[fv],
            schema=[Field(name="first_char", dtype=String)],
        )
        def post_changed(inputs: pd.DataFrame) -> pd.DataFrame:
            df = pd.DataFrame()
            df["first_char"] = inputs["string_col"].str[:1].astype("string") + "hi"
            return df

        feast_object_diffs = diff_registry_objects(
            pre_changed, pre_changed, "on demand feature view"
        )
        assert len(feast_object_diffs.feast_object_property_diffs) == 0

        feast_object_diffs = diff_registry_objects(
            pre_changed, post_changed, "on demand feature view"
        )

        # Note that user_defined_function.body is excluded because it always changes (dill is non-deterministic), even
        # if no code is changed
        assert len(feast_object_diffs.feast_object_property_diffs) == 3
        assert feast_object_diffs.feast_object_property_diffs[0].property_name == "name"
        # Note we should only now be looking at changes for the feature_transformation field
        assert (
            feast_object_diffs.feast_object_property_diffs[1].property_name
            == "feature_transformation.name"
        )
        assert (
            feast_object_diffs.feast_object_property_diffs[2].property_name
            == "feature_transformation.body_text"
        )


def test_diff_registry_objects_batch_to_push_source(simple_dataset_1):