def test_remove_null_rows_min_negative_unchanged(self): df = pd.DataFrame([[0, 1, -2], [1, 2, 3], [0, 0, 0], [4, 5, 6]]) df2 = preprocess.remove_null_rows(df) self.assertEqual(4, df2.shape[0], "wrong number of rows")
def test_remove_null_rows_min_0_remove_ok(self): df = pd.DataFrame([[0, 1, 2], [1, 2, 3], [0, 0, 0], [4, 5, 6]]) df2 = preprocess.remove_null_rows(df) self.assertEqual(3, df2.shape[0], "wrong number of rows")