Removed unnecessary "sources" in Python SDK unit tests

feast-dev · Nov 19, 2019 · 2d8cc6f · 2d8cc6f
1 parent a75116c
commit 2d8cc6f
Showing 1 changed file with 0 additions and 14 deletions.
diff --git a/sdk/python/tests/test_client.py b/sdk/python/tests/test_client.py
@@ -367,11 +367,6 @@ def test_feature_set_ingest_success(self, dataframe, client, mocker):
         driver_fs.add(Feature(name="feature_3", dtype=ValueType.INT64))
         driver_fs.add(Entity(name="entity_id", dtype=ValueType.INT64))
 
-        driver_fs.source = KafkaSource(topic="feature-topic", brokers="127.0.0.1")
-
-        client._message_producer = MagicMock()
-        client._message_producer.produce = MagicMock()
-
         # Register with Feast core
         client.apply(driver_fs)
 
@@ -397,11 +392,6 @@ 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"
-            )
-            client._message_producer = MagicMock()
-            client._message_producer.produce = MagicMock()
 
             # Update based on dataset
             driver_fs.infer_fields_from_df(dataframe)
@@ -437,10 +427,6 @@ def test_feature_set_types_success(self, client, dataframe, mocker):
             max_age=Duration(seconds=3600),
         )
 
-        all_types_fs.source = KafkaSource(topic="feature-topic", brokers="127.0.0.1")
-        client._message_producer = MagicMock()
-        client._message_producer.produce = MagicMock()
-
         # Register with Feast core
         client.apply(all_types_fs)