def test_update_from_source_success(self, dataframe): fs = FeatureSet("driver-feature-set") fs.update_from_dataset( dataframe, column_mapping={ "entity_id": Entity(name="entity", dtype=ValueType.INT64) }, ) assert len(fs.features) == 3 and fs.features[1].name == "feature_2"
def test_feature_set_ingest_failure(self, client, dataframe, exception): with pytest.raises(exception): # Create feature set driver_fs = FeatureSet("driver-feature-set") driver_fs.source = KafkaSource(topic="feature-topic", brokers="fake.broker.com") driver_fs._message_producer = MagicMock() driver_fs._message_producer.send = MagicMock() # Update based on dataset driver_fs.update_from_dataset( dataframe, column_mapping={ "entity_id": Entity(name="entity", dtype=ValueType.INT64) }, ) # Register with Feast core client.apply(driver_fs) # Ingest data into Feast driver_fs.ingest(dataframe=dataframe)
def test_update_from_source_failure(self): with pytest.raises(Exception): df = pd.DataFrame() fs = FeatureSet("driver-feature-set") fs.update_from_dataset(df)