Fix test_compute

lsst-sqre · Oct 3, 2021 · d866934 · d866934
1 parent 3a2f274
commit d866934
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/tests/kafkaaggregator/test_compute.py b/tests/kafkaaggregator/test_compute.py
@@ -6,6 +6,7 @@
 import pytest
 
 from kafkaaggregator.aggregator import Aggregator
+from kafkaaggregator.aggregator_config import AggregatorConfig
 from kafkaaggregator.fields import Field
 from kafkaaggregator.models import create_record
 
@@ -90,10 +91,11 @@ def test_compute(
     expected_result: `dict`
         Dictionary with the expected result for the aggregated_message
     """
-    Agg = Aggregator(
-        config_file=config_dir.joinpath("aggregator_config.yaml"),
-        aggregated_topic="aggregated_example0",
-    )
+    config_file = config_dir.joinpath("aggregator_config.yaml")
+
+    aggregated_topic = AggregatorConfig(config_file).get("aggregated_example0")
+
+    Agg = Aggregator(aggregated_topic)
 
     # Mock the creation of the aggregated fields
     Agg._aggregated_fields = aggregated_fields
@@ -132,12 +134,11 @@ def test_compute_min_sample_size(
     expected_result: `dict`
         Dictionary with the expected result for the aggregated_message
     """
-    Agg = Aggregator(
-        config_file=config_dir.joinpath(
-            "aggregator_config_min_sample_size.yaml"
-        ),
-        aggregated_topic="aggregated_example0",
-    )
+    config_file = config_dir.joinpath("aggregator_config_min_sample_size.yaml")
+
+    aggregated_topic = AggregatorConfig(config_file).get("aggregated_example0")
+
+    Agg = Aggregator(aggregated_topic)
 
     # Mock the creation of the aggregated fields
     Agg._aggregated_fields = aggregated_fields