def test_invalidRows2(): testing_df = df.copy() testing_df['col3'] = pd.Series(['1'], index = [0]) result_df = testing_df.copy() return_df = load.rowsWithInvalidValuesInColumns(testing_df, [0, 2]) result_df.drop(result_df.index[0], inplace=True) assert (((return_df.fillna(0) == result_df.fillna(0)).all()).all()) == True
def test_invalidRows1(): testing_df = df.copy() result_df = df.copy() return_df = load.rowsWithInvalidValuesInColumns(testing_df, [0]) result_df.drop(result_df.index[0], inplace=True) assert (((return_df.fillna(0) == result_df.fillna(0)).all()).all()) == True