Skip to content
This repository was archived by the owner on Apr 11, 2025. It is now read-only.

Commit a233ea4

Browse files
sverma25bosd
authored andcommitted
Added test cases
1 parent 3d3efd4 commit a233ea4

File tree

2 files changed

+30
-0
lines changed

2 files changed

+30
-0
lines changed

tests/data.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3838,3 +3838,20 @@
38383838
38393839
],
38403840
]
3841+
3842+
data_multi_params1 = [
3843+
["Number of Coils", "Number of Paperclips"],
3844+
["5", "3, 5, 4"],
3845+
["10", "7, 8, 6"],
3846+
["15", "11, 10, 12"],
3847+
["20", "15, 13, 14"]
3848+
]
3849+
3850+
data_multi_params2 = [
3851+
["Time (drops of water)", "Distance (cm)"],
3852+
["1", "10,11,9"],
3853+
["2", "29, 31, 30"],
3854+
["3", "59, 58, 61"],
3855+
["4", "102, 100, 98"],
3856+
["5", "122, 125, 127"]
3857+
]

tests/test_common.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,3 +214,16 @@ def _make_table(page, order):
214214
assert iterator_b is not None
215215
item_c = next(iterator_b)
216216
assert item_c is not None
217+
218+
def test_multi_params(testdir):
219+
df1 = pd.DataFrame(data_multi_params1)
220+
df2 = pd.DataFrame(data_multi_params2)
221+
222+
filename = os.path.join(
223+
testdir, "multi_params.pdf"
224+
)
225+
tables = camelot.read_pdf(filename, pages="all", multi={'2': {"table_regions": ["120, 210, 400, 90"]}},
226+
split_text=True)
227+
228+
assert df1.equals(tables[0].df)
229+
assert df2.equals(tables[1].df)

0 commit comments

Comments
 (0)