Add additional test cases

googleapis · Oct 23, 2023 · cc3556a · cc3556a
1 parent f3b7434
commit cc3556a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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):