diff --git a/tests/unit/test_page.py b/tests/unit/test_page.py index 0201180..0883216 100644 --- a/tests/unit/test_page.py +++ b/tests/unit/test_page.py @@ -262,8 +262,14 @@ def test_page_elements_large_document(large_docproto): wrapped_page = page.Page( documentai_object=pg, document_text=large_docproto.text ) + for block in wrapped_page.blocks: + assert block.text != "" for paragraph in wrapped_page.paragraphs: assert paragraph.text != "" + for line in wrapped_page.lines: + assert line.text != "" + for token in wrapped_page.tokens: + assert token.text != "" def test_Line(docproto):