style(unit_test): avoid if continue

dbt-labs · Nov 11, 2024 · 0afd0d9 · 0afd0d9
1 parent 64bec4d
commit 0afd0d9
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/core/dbt/parser/unit_tests.py b/core/dbt/parser/unit_tests.py
@@ -158,12 +158,15 @@ def parse_unit_test_case(self, test_case: UnitTestDefinition):
                     defer_relation=original_input_node.defer_relation,
                 )
 
-                if not resource_type == NodeType.Model:
-                    continue
                 if original_input_node.version:
                     input_node.version = original_input_node.version
                 if original_input_node.latest_version:
                     input_node.latest_version = original_input_node.latest_version
+            elif resource_type in (NodeType.Seed, NodeType.Snapshot):
+                input_node = ModelNode(
+                    **common_fields,
+                    defer_relation=original_input_node.defer_relation,
+                )
 
             elif resource_type == NodeType.Source:
                 # We are reusing the database/schema/identifier from the original source,