def test_add_rows_noop(seed, df):
num = 0
result = syn._remove_rows(df, num)
assert len(result) == len(df) + num
pdt.assert_frame_equal(result, df)
def test_add_rows_noop(seed, df):
num = 0
result = syn._remove_rows(df, num)
assert len(result) == len(df) + num
pdt.assert_frame_equal(result, df)
def test_remove_rows(seed, df, alloc_id, count):
num = 2
result = syn._remove_rows(df, num)
assert len(result) == len(df) - num
pdt.assert_frame_equal(
result,
pd.DataFrame(
{alloc_id: ['b', 'c', 'b'],
count: [2, 3, 4]},
index=[1, 2, 3]))
def test_remove_rows(seed, df, alloc_id, count):
num = 2
result = syn._remove_rows(df, num)
assert len(result) == len(df) - num
pdt.assert_frame_equal(
result,
pd.DataFrame({
alloc_id: ['b', 'c', 'b'],
count: [2, 3, 4]
},
index=[1, 2, 3]))