fix tests

dbt-labs · Nov 12, 2024 · bf941a3 · bf941a3
1 parent 76379bd
commit bf941a3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/tests/functional/metrics/test_metrics.py b/tests/functional/metrics/test_metrics.py
@@ -85,7 +85,7 @@ def test_simple_metric(
         )
         assert (
             manifest.metrics["metric.test.number_of_people"].time_granularity
-            == TimeGranularity.MONTH
+            == TimeGranularity.MONTH.value
         )
         assert manifest.metrics["metric.test.collective_tenure"].time_granularity is None
 
@@ -436,25 +436,25 @@ def test_cumulative_metric(self, project):
         metric_ids = set(manifest.metrics.keys())
         expected_metric_ids_to_cumulative_type_params = {
             "metric.test.weekly_visits": CumulativeTypeParams(
-                window=MetricTimeWindow(count=7, granularity=TimeGranularity.DAY),
+                window=MetricTimeWindow(count=7, granularity=TimeGranularity.DAY.value),
                 period_agg=PeriodAggregation.AVERAGE,
             ),
             "metric.test.cumulative_orders": CumulativeTypeParams(
                 period_agg=PeriodAggregation.LAST
             ),
             "metric.test.orders_ytd": CumulativeTypeParams(
-                grain_to_date=TimeGranularity.YEAR, period_agg=PeriodAggregation.FIRST
+                grain_to_date=TimeGranularity.YEAR.value, period_agg=PeriodAggregation.FIRST
             ),
             "metric.test.monthly_orders": CumulativeTypeParams(
-                window=MetricTimeWindow(count=1, granularity=TimeGranularity.MONTH),
+                window=MetricTimeWindow(count=1, granularity=TimeGranularity.MONTH.value),
                 period_agg=PeriodAggregation.AVERAGE,
             ),
             "metric.test.yearly_orders": CumulativeTypeParams(
-                window=MetricTimeWindow(count=1, granularity=TimeGranularity.YEAR),
+                window=MetricTimeWindow(count=1, granularity=TimeGranularity.YEAR.value),
                 period_agg=PeriodAggregation.FIRST,
             ),
             "metric.test.visits_mtd": CumulativeTypeParams(
-                grain_to_date=TimeGranularity.MONTH, period_agg=PeriodAggregation.FIRST
+                grain_to_date=TimeGranularity.MONTH.value, period_agg=PeriodAggregation.FIRST
             ),
             "metric.test.cumulative_visits": CumulativeTypeParams(
                 period_agg=PeriodAggregation.FIRST

diff --git a/tests/unit/test_semantic_layer_nodes_satisfy_protocols.py b/tests/unit/test_semantic_layer_nodes_satisfy_protocols.py
@@ -196,7 +196,7 @@ def where_filter() -> WhereFilter:
 
 @pytest.fixture(scope="session")
 def metric_time_window() -> MetricTimeWindow:
-    return MetricTimeWindow(count=1, granularity=TimeGranularity.DAY)
+    return MetricTimeWindow(count=1, granularity=TimeGranularity.DAY.value)
 
 
 @pytest.fixture(scope="session")
@@ -211,7 +211,7 @@ def complex_metric_input(metric_time_window, where_filter) -> MetricInput:
         filter=where_filter,
         alias="aliased_metric_input",
         offset_window=metric_time_window,
-        offset_to_grain=TimeGranularity.DAY,
+        offset_to_grain=TimeGranularity.DAY.value,
     )