def test_add_rows(seed, df, alloc_id, count, constraint):
num = 3
result = syn._add_rows(df, num, alloc_id, constraint)
assert len(result) == len(df) + num
pdt.assert_frame_equal(
result,
pd.DataFrame(
{alloc_id: ['a', 'b', 'c', 'b', 'c', 'b', 'c', 'c'],
count: [1, 2, 3, 4, 5, 5, 1, 4]}))
def test_add_rows_no_stuff(seed, df, alloc_id, count, constraint):
num = 5
stuff = False
result = syn._add_rows(df, num, alloc_id, constraint, stuff=stuff)
assert len(result) == len(df) + num
pdt.assert_frame_equal(
result,
pd.DataFrame(
{alloc_id: ['a', 'b', 'c', 'b', 'c', 'b', 'c', 'c', 'c', None],
count: [1, 2, 3, 4, 5, 5, 1, 4, 4, 4]}))
def test_add_rows(seed, df, alloc_id, count, constraint):
num = 3
result = syn._add_rows(df, num, alloc_id, constraint)
assert len(result) == len(df) + num
pdt.assert_frame_equal(
result,
pd.DataFrame({
alloc_id: ['a', 'b', 'c', 'b', 'c', 'b', 'c', 'c'],
count: [1, 2, 3, 4, 5, 5, 1, 4]
}))
def test_add_rows_no_stuff(seed, df, alloc_id, count, constraint):
num = 5
stuff = False
result = syn._add_rows(df, num, alloc_id, constraint, stuff=stuff)
assert len(result) == len(df) + num
pdt.assert_frame_equal(
result,
pd.DataFrame({
alloc_id: ['a', 'b', 'c', 'b', 'c', 'b', 'c', 'c', 'c', None],
count: [1, 2, 3, 4, 5, 5, 1, 4, 4, 4]
}))
