Skip to content

Commit

Permalink
edited comments
Browse files Browse the repository at this point in the history
  • Loading branch information
arinkulshi committed Apr 25, 2024
1 parent e01186a commit 0dbb0cc
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions OCR/tests/pdf_field_extractor_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import os


# Setup a fixture for the PDF extractor

@pytest.fixture
def pdf_extractor():
current_script_dir = os.path.dirname(os.path.abspath(__file__))
Expand All @@ -17,12 +17,10 @@ def pdf_extractor():


def test_initialize_reader(pdf_extractor):
# Check if the reader is initialized
assert pdf_extractor.reader is not None, "The PDF reader should be initialized."


def test_close_reader(pdf_extractor):
# Close the reader and check if it's closed properly
pdf_extractor.close_reader()
assert pdf_extractor.reader is None, "The PDF reader should be closed."

Expand All @@ -35,7 +33,6 @@ def test_segment_fields(pdf_extractor):
# Expected coordinates for the "Address" field
expected_address_rect = [80.08, 611.28, 339.76, 624.151]

# Find the "Address" field in the form_fields list
address_field = None
for field_name, rect in pdf_extractor.form_fields:
if field_name == "Address":
Expand Down

0 comments on commit 0dbb0cc

Please sign in to comment.