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] }))