fix unit tests

dbt-labs · Sep 30, 2024 · 7e9ab1c · 7e9ab1c
1 parent 1422944
commit 7e9ab1c
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git a/tests/unit/contracts/graph/test_nodes.py b/tests/unit/contracts/graph/test_nodes.py
@@ -1,3 +1,4 @@
+from argparse import Namespace
 import pickle
 import re
 from dataclasses import replace
@@ -24,6 +25,13 @@
 )
 
 
+@pytest.fixture
+def args_for_flags() -> Namespace:
+    return Namespace(
+        state_modified_compare_vars=False
+    )
+
+
 def norm_whitespace(string):
     _RE_COMBINE_WHITESPACE = re.compile(r"\s+")
     string = _RE_COMBINE_WHITESPACE.sub(" ", string).strip()

diff --git a/tests/unit/contracts/graph/test_nodes_parsed.py b/tests/unit/contracts/graph/test_nodes_parsed.py
@@ -65,8 +65,10 @@
 
 
 @pytest.fixture
-def flags_for_args() -> Namespace:
-    return Namespace(SEND_ANONYMOUS_USAGE_STATS=False)
+def args_for_flags() -> Namespace:
+    return Namespace(
+        send_anonymous_usage_stats=False, state_modified_compare_vars=False
+    )
 
 
 @pytest.fixture

diff --git a/tests/unit/graph/test_selector_methods.py b/tests/unit/graph/test_selector_methods.py
@@ -1,3 +1,4 @@
+from argparse import Namespace
 import copy
 from dataclasses import replace
 from pathlib import Path
@@ -643,6 +644,10 @@ def previous_state(manifest):
     return create_previous_state(manifest)
 
 
+@pytest.fixture
+def args_for_flags():
+    return Namespace(state_modified_compare_vars=False)
+
 def add_node(manifest, node):
     manifest.nodes[node.unique_id] = node