Moses Paul R
commited on
Commit
·
fd6841d
1
Parent(s):
fcb3e12
fix tests
Browse files
tests/builders/test_rotated_bboxes.py
CHANGED
|
@@ -11,7 +11,7 @@ def test_rotated_bboxes(pdf_document):
|
|
| 11 |
# Ensure we match all text lines up properly
|
| 12 |
text_lines = first_page.contained_blocks(pdf_document, (BlockTypes.Line,))
|
| 13 |
text_blocks = first_page.contained_blocks(pdf_document, (BlockTypes.Text,))
|
| 14 |
-
assert len(text_lines) ==
|
| 15 |
|
| 16 |
# Ensure the bbox sizes match up
|
| 17 |
max_line_position = max([line.polygon.x_end for line in text_lines])
|
|
|
|
| 11 |
# Ensure we match all text lines up properly
|
| 12 |
text_lines = first_page.contained_blocks(pdf_document, (BlockTypes.Line,))
|
| 13 |
text_blocks = first_page.contained_blocks(pdf_document, (BlockTypes.Text,))
|
| 14 |
+
assert len(text_lines) == 84
|
| 15 |
|
| 16 |
# Ensure the bbox sizes match up
|
| 17 |
max_line_position = max([line.polygon.x_end for line in text_lines])
|
tests/providers/test_pdf_provider.py
CHANGED
|
@@ -8,7 +8,7 @@ def test_pdf_provider(pdf_provider):
|
|
| 8 |
assert pdf_provider.get_images([0], 96)[0].size == (816, 1056)
|
| 9 |
|
| 10 |
page_lines = pdf_provider.get_page_lines(0)
|
| 11 |
-
assert len(page_lines) ==
|
| 12 |
|
| 13 |
spans = page_lines[0].spans
|
| 14 |
assert len(spans) == 2
|
|
|
|
| 8 |
assert pdf_provider.get_images([0], 96)[0].size == (816, 1056)
|
| 9 |
|
| 10 |
page_lines = pdf_provider.get_page_lines(0)
|
| 11 |
+
assert len(page_lines) == 85
|
| 12 |
|
| 13 |
spans = page_lines[0].spans
|
| 14 |
assert len(spans) == 2
|