Feature test increase ok method on check (#327)

* Added checks for ok method in check

* Added ok boolean method for check

cuallee/polars_validation.py

@@ -530,6 +530,7 @@ def _evaluate_status(pass_rate, pass_threshold):
     pl.Config.set_tbl_cols(12)
     return pl.DataFrame(computation_basis)
 
+
 def ok(check: Check, dataframe: pl.DataFrame) -> bool:
     """True when all rules in the check pass validation"""
 
@@ -540,4 +541,4 @@ def ok(check: Check, dataframe: pl.DataFrame) -> bool:
         operator.methodcaller("to_series"),
         operator.methodcaller("select", "status"),
     )
-    return _all_pass(summary(check, dataframe))
+    return _all_pass(summary(check, dataframe))

test/unit/class_check/test_ok.py

@@ -0,0 +1,24 @@
+import pytest
+from cuallee import Check
+import pandas as pd
+import polars as pl
+import duckdb as dk
+from pyspark.sql import SparkSession
+
+
+def test_pandas_ok(check: Check):
+    df = pd.DataFrame({"A": range(10)})
+    check.is_complete("A")
+    assert check.ok(df)
+
+
+def test_polars_ok(check: Check):
+    df = pl.DataFrame({"A": range(10)})
+    check.is_complete("A")
+    assert check.ok(df)
+
+
+def test_pyspark_ok(check: Check, spark: SparkSession):
+    df = spark.range(10)
+    check.is_complete("id")
+    assert check.ok(df)