def test_delete_na1(): testing_df = df.copy() result_df = df.copy() clean.deleteRowsWithNA(testing_df, 0) result_df.dropna(subset=['col1'], inplace=True) result_df.reset_index(drop=True, inplace=True) assert (((testing_df.fillna(0) == result_df.fillna(0)).all()).all()) == True
def test_delete_na3(): testing_df = df.copy() with pytest.raises(Exception): clean.deleteRowsWithNA(testing_df, 16)
def test_delete_na2(): testing_df = df.copy() result_df = df.copy() clean.deleteRowsWithNA(testing_df, 2) assert (((testing_df.fillna(0) == result_df.fillna(0)).all()).all()) == True