def test_remove_rows_by_count_noop(df, count): amount = 0 result = syn._remove_rows_by_count(df, amount, count) assert result[count].sum() == df[count].sum() pdt.assert_frame_equal(result, df)
def test_remove_rows_by_count(df, count, alloc_id): amount = 10 result = syn._remove_rows_by_count(df, amount, count) assert result[count].sum() == df[count].sum() - amount pdt.assert_frame_equal( result, pd.DataFrame( {alloc_id: ['b', 'c'], count: [2, 3]}, index=[1, 2]))
def test_remove_rows_by_count(df, count, alloc_id): amount = 10 result = syn._remove_rows_by_count(df, amount, count) assert result[count].sum() == df[count].sum() - amount pdt.assert_frame_equal( result, pd.DataFrame({ alloc_id: ['b', 'c'], count: [2, 3] }, index=[1, 2]))