Skip to content

Commit 05d7bf3

Browse files
Conditionally get test metadata
1 parent 9f6aa67 commit 05d7bf3

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/dbt_score/models.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,8 @@ def from_node(cls, test_node: dict[str, Any]) -> "Test":
6363
"""Create a test object from a test node in the manifest."""
6464
return cls(
6565
name=test_node["name"],
66-
type=test_node["test_metadata"]["name"],
67-
kwargs=test_node["test_metadata"].get("kwargs", {}),
66+
type=test_node.get("test_metadata", {}).get("name", "generic"),
67+
kwargs=test_node.get("test_metadata", {}).get("kwargs", {}),
6868
tags=test_node.get("tags", []),
6969
_raw_values=test_node,
7070
)
@@ -359,7 +359,9 @@ def from_node(
359359
tests=[
360360
Test.from_node(test)
361361
for test in test_values
362-
if not test["test_metadata"]["kwargs"].get("column_name")
362+
if not test.get("test_metadata", {}) # Not all tests have metadata.
363+
.get("kwargs", {})
364+
.get("column_name")
363365
],
364366
_raw_values=node_values,
365367
_raw_test_values=test_values,

0 commit comments

Comments
 (0)