Check tests: update run_check return type to List[Row].

nikie · nikie · commit ecad4d1039a2 · 2026-05-04T23:14:40.000+03:00
diff --git a/tests/test_checks.py b/tests/test_checks.py
@@ -94,7 +94,7 @@ def test_initializer(self):
             ),
         )
 
-    def run_check(self, check: Check, columns: Union[str, List[str]] = "constraint_status") -> list[Row]:
+    def run_check(self, check: Check, columns: Union[str, List[str]] = "constraint_status") -> List[Row]:
         columns = [columns] if isinstance(columns, str) else columns
         result = VerificationSuite(self.spark).onData(self.df).addCheck(check).run()
         df = VerificationResult.checkResultsAsDataFrame(self.spark, result)

Original file line number	Diff line number	Diff line change
`@@ -94,7 +94,7 @@ def test_initializer(self):`
`94`	`94`	`),`
`95`	`95`	`)`
`96`	`96`
`97`		`- def run_check(self, check: Check, columns: Union[str, List[str]] = "constraint_status") -> list[Row]:`
	`97`	`+ def run_check(self, check: Check, columns: Union[str, List[str]] = "constraint_status") -> List[Row]:`
`98`	`98`	`columns = [columns] if isinstance(columns, str) else columns`
`99`	`99`	`result = VerificationSuite(self.spark).onData(self.df).addCheck(check).run()`
`100`	`100`	`df = VerificationResult.checkResultsAsDataFrame(self.spark, result)`