@@ -28,14 +28,7 @@ def test_distinct_test_results():
2828 test_result_ids = set ()
2929 for test_results in report_data ["test_results" ].values ():
3030 for test_result in test_results :
31- metadata = test_result ["metadata" ]
32- test_result_id = (
33- metadata ["test_unique_id" ],
34- metadata ["table_name" ],
35- metadata ["column_name" ],
36- metadata ["test_type" ],
37- metadata ["test_sub_type" ],
38- )
31+ test_result_id = get_test_result_id (test_result ["metadata" ])
3932 assert test_result_id not in test_result_ids
4033 test_result_ids .add (test_result_id )
4134
@@ -77,6 +70,28 @@ def test_sidebar(report_data_fixture):
7770 )
7871
7972
73+ def get_test_result_id (test_metadata ):
74+ test_type_id_map = {
75+ "anomaly_detection" : (
76+ test_metadata ["model_unique_id" ],
77+ test_metadata ["column_name" ],
78+ test_metadata ["test_unique_id" ],
79+ test_metadata ["test_sub_type" ],
80+ ),
81+ "schema_change" : (
82+ test_metadata ["model_unique_id" ],
83+ test_metadata ["test_unique_id" ],
84+ ),
85+ "dbt_test" : (
86+ test_metadata ["model_unique_id" ],
87+ test_metadata ["column_name" ],
88+ test_metadata ["test_unique_id" ],
89+ test_metadata ["test_sub_type" ],
90+ ),
91+ }
92+ return test_type_id_map [test_metadata ["test_type" ]]
93+
94+
8095def assert_totals (data_totals : Totals , fixture_totals : Totals ):
8196 assert data_totals .keys () == fixture_totals .keys ()
8297 for total_key in fixture_totals :
0 commit comments