def test_removeRows2(): testing_df = df.copy() result_df = df.copy() load.removeRows(testing_df, [0, 1]) result_df.drop(result_df.index[0 : 2], inplace=True) result_df.reset_index(drop=True, inplace=True) assert (((testing_df.fillna(0) == result_df.fillna(0)).all()).all()) == True
def test_removeRow3(): with pytest.raises(Exception): load.removeRows(testing_df, [6, 12])