Skip to content

Commit

Permalink
Avoid division by 0
Browse files Browse the repository at this point in the history
  • Loading branch information
gabriel-piles committed Dec 17, 2024
1 parent a296fcb commit c3ad18b
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 3 deletions.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[project]
name = "pdf-document-layout-analysis"
version = "2024.12.17.5"
version = "2024.12.17.7"
description = "This tool is for PDF document layout analysis"
license = { file = "LICENSE" }
authors = [{ name = "HURIDOCS" }]
Expand Down
3 changes: 3 additions & 0 deletions src/pdf_features/Rectangle.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,9 @@ def from_poppler_tag_etree(tag: ElementBase) -> "Rectangle":
x_max = x_min + int(tag.attrib["width"])
y_max = y_min + int(tag.attrib["height"])

if len(content) <= 1:
return Rectangle(x_min, y_min, x_max, y_max)

one_character_length = max(int((x_max - x_min) / len(content)), 2)
if content[0] == " ":
x_min += one_character_length
Expand Down
4 changes: 2 additions & 2 deletions src/test_end_to_end.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ def test_regular_pdf(self):
self.assertEqual(expected_content, results_list[0]["text"])
self.assertEqual(157, results_list[0]["left"])
self.assertEqual(105, results_list[0]["top"])
self.assertEqual(283, results_list[0]["width"])
self.assertEqual(282, results_list[0]["width"])
self.assertEqual(36, results_list[0]["height"])
self.assertEqual(1, results_list[0]["page_number"])
self.assertEqual(595, results_list[0]["page_width"])
Expand Down Expand Up @@ -112,7 +112,7 @@ def test_regular_pdf_fast(self):
self.assertEqual(expected_content, results_list[0]["text"])
self.assertEqual(157, results_list[0]["left"])
self.assertEqual(106, results_list[0]["top"])
self.assertEqual(284, results_list[0]["width"])
self.assertEqual(278, results_list[0]["width"])
self.assertEqual(24, results_list[0]["height"])
self.assertEqual(1, results_list[0]["page_number"])
self.assertEqual(595, results_list[0]["page_width"])
Expand Down

0 comments on commit c3ad18b

Please sign in to comment.