def test_pandas_df_missing_rows(self) -> None:
missing_df = pd.DataFrame({
"x1": [2, None],
"x2": [None, "str4"]
})
output_df = has_missing(self.pd_df_with_missing, return_missing=True).reset_index(drop=True)
self.assertEqual(missing_df.equals(output_df), True)
def test_spark_df_without_missing(self) -> None:
self.assertEqual(has_missing(self.spark_df_without_missing), False)
def test_spark_df_with_missing(self) -> None:
self.assertEqual(has_missing(self.spark_df_with_missing), True)
def test_pandas_df_without_missing(self) -> None:
self.assertEqual(has_missing(self.pd_df_without_missing), False)
def test_pandas_df_with_missing(self) -> None:
self.assertEqual(has_missing(self.pd_df_with_missing), True)