add check_obj as an arg and small improvements

Signed-off-by: Filipe Oliveira <[email protected]>
unionai-oss · Nov 13, 2023 · 952bef3 · 952bef3
1 parent ee0e42c
commit 952bef3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/pyspark/test_pyspark_config.py b/tests/pyspark/test_pyspark_config.py
@@ -47,8 +47,8 @@ class TestSchema(DataFrameModel):
         }
 
         assert CONFIG.dict() == expected
-        assert pandera_schema.validate(input_df) is not None
-        assert TestSchema.validate(input_df) is not None
+        assert pandera_schema.validate(input_df)
+        assert TestSchema.validate(input_df)
 
     # pylint:disable=too-many-locals
     def test_schema_only(self, spark, sample_spark_schema):