Get test_metadata optionally everywhere

PicnicSupermarket · Dec 16, 2024 · 6df3ba8 · 6df3ba8
1 parent 05d7bf3
commit 6df3ba8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/dbt_score/models.py b/src/dbt_score/models.py
@@ -141,7 +141,10 @@ def _get_columns(
                 [
                     test
                     for test in test_values
-                    if test["test_metadata"]["kwargs"].get("column_name") == name
+                    if test.get("test_metadata", {})
+                    .get("kwargs", {})
+                    .get("column_name")
+                    == name
                 ],
             )
             for name, values in node_values.get("columns", {}).items()
@@ -224,7 +227,9 @@ def from_node(
             tests=[
                 Test.from_node(test)
                 for test in test_values
-                if not test["test_metadata"]["kwargs"].get("column_name")
+                if not test.get("test_metadata", {})
+                .get("kwargs", {})
+                .get("column_name")
             ],
             depends_on=node_values["depends_on"],
             _raw_values=node_values,