Briq
/
camelot-py
6
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Further linting fixes flagged by DeepSource

pull/127/head
Frh 2020-04-06 18:41:10 -07:00
parent f1d4a32bc6
commit dff9f5cd82
2 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/data.py
View File

@ -838,7 +838,8 @@ data_stream_two_tables_1 = [
"2,330 .9", "2,330 .9",
], ],
[ [
"Violent crime . . . . . . . .\n . .\n . .\n . .\n . .\n . .", "Violent crime . . . . . . . .\n . .\n . .\n . .\n" \
" . .\n . .",
"467 .9", "467 .9",
"69 .1", "69 .1",
"398 .8", "398 .8",

17
tests/test_common.py
View File

@ -180,6 +180,8 @@ def test_stream_layout_kwargs():
def test_stream_vertical_header(): def test_stream_vertical_header():
"""Tests a complex table inclusive of vertically aligned header.
"""
df = pd.DataFrame(data_stream_vertical_headers) df = pd.DataFrame(data_stream_vertical_headers)
filename = os.path.join(testdir, "vertical_header.pdf") filename = os.path.join(testdir, "vertical_header.pdf")
@ -274,8 +276,7 @@ def test_repr():
assert repr(tables) == "<TableList n=1>" assert repr(tables) == "<TableList n=1>"
assert repr(tables[0]) == "<Table shape=(7, 7)>" assert repr(tables[0]) == "<Table shape=(7, 7)>"
assert ( assert (
repr(tables[0].cells[0][0]) repr(tables[0].cells[0][0]) == "<Cell x1=120.48 y1=218.43 x2=164.64 y2=233.77>"
== "<Cell x1=120.48 y1=218.43 x2=164.64 y2=233.77>"
) )
@ -285,24 +286,21 @@ def test_pages():
assert repr(tables) == "<TableList n=1>" assert repr(tables) == "<TableList n=1>"
assert repr(tables[0]) == "<Table shape=(7, 7)>" assert repr(tables[0]) == "<Table shape=(7, 7)>"
assert ( assert (
repr(tables[0].cells[0][0]) repr(tables[0].cells[0][0]) == "<Cell x1=120.48 y1=218.43 x2=164.64 y2=233.77>"
== "<Cell x1=120.48 y1=218.43 x2=164.64 y2=233.77>"
) )
tables = camelot.read_pdf(url, pages="1-end") tables = camelot.read_pdf(url, pages="1-end")
assert repr(tables) == "<TableList n=1>" assert repr(tables) == "<TableList n=1>"
assert repr(tables[0]) == "<Table shape=(7, 7)>" assert repr(tables[0]) == "<Table shape=(7, 7)>"
assert ( assert (
repr(tables[0].cells[0][0]) repr(tables[0].cells[0][0]) == "<Cell x1=120.48 y1=218.43 x2=164.64 y2=233.77>"
== "<Cell x1=120.48 y1=218.43 x2=164.64 y2=233.77>"
) )
tables = camelot.read_pdf(url, pages="all") tables = camelot.read_pdf(url, pages="all")
assert repr(tables) == "<TableList n=1>" assert repr(tables) == "<TableList n=1>"
assert repr(tables[0]) == "<Table shape=(7, 7)>" assert repr(tables[0]) == "<Table shape=(7, 7)>"
assert ( assert (
repr(tables[0].cells[0][0]) repr(tables[0].cells[0][0]) == "<Cell x1=120.48 y1=218.43 x2=164.64 y2=233.77>"
== "<Cell x1=120.48 y1=218.43 x2=164.64 y2=233.77>"
) )
@ -312,8 +310,7 @@ def test_url():
assert repr(tables) == "<TableList n=1>" assert repr(tables) == "<TableList n=1>"
assert repr(tables[0]) == "<Table shape=(7, 7)>" assert repr(tables[0]) == "<Table shape=(7, 7)>"
assert ( assert (
repr(tables[0].cells[0][0]) repr(tables[0].cells[0][0]) == "<Cell x1=120.48 y1=218.43 x2=164.64 y2=233.77>"
== "<Cell x1=120.48 y1=218.43 x2=164.64 y2=233.77>"
) )