Fix test helpers.

dbt-labs · Nov 13, 2024 · dfec513 · dfec513
1 parent 5f14ab2
commit dfec513
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/metricflow-semantics/metricflow_semantics/test_helpers/snapshot_helpers.py b/metricflow-semantics/metricflow_semantics/test_helpers/snapshot_helpers.py
@@ -426,6 +426,7 @@ def assert_str_snapshot_equal(  # noqa: D103
     snapshot_id: str,
     snapshot_str: str,
     expectation_description: Optional[str] = None,
+    incomparable_strings_replacement_function: Optional[Callable[[str], str]] = None,
 ) -> None:
     """Write / compare a string snapshot."""
     assert_snapshot_text_equal(
@@ -436,4 +437,5 @@ def assert_str_snapshot_equal(  # noqa: D103
         snapshot_text=snapshot_str,
         snapshot_file_extension=".txt",
         expectation_description=expectation_description,
+        incomparable_strings_replacement_function=incomparable_strings_replacement_function,
     )
diff --git a/tests_metricflow/sql/compare_sql_plan.py b/tests_metricflow/sql/compare_sql_plan.py
@@ -27,7 +27,6 @@ def assert_default_rendered_sql_equal(
 ) -> None:
     """Helper function to render a select statement and compare with the one saved as a file."""
     sql_query_plan = SqlQueryPlan(render_node=sql_plan_node, plan_id=DagId.from_str(plan_id))
-
     rendered_sql = DefaultSqlQueryPlanRenderer().render_sql_query_plan(sql_query_plan).sql
 
     assert_plan_snapshot_text_equal(