refactor: revert embedded_text_region unit test

Unstructured-IO · May 20, 2024 · 3d3b024 · 3d3b024
1 parent 5e48003
commit 3d3b024
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test_unstructured_inference/inference/test_layout.py b/test_unstructured_inference/inference/test_layout.py
@@ -409,6 +409,12 @@ def check_annotated_image():
         check_annotated_image()
 
 
+@pytest.mark.parametrize(("text", "expected"), [("asdf", "asdf"), (None, "")])
+def test_embedded_text_region(text, expected):
+    etr = elements.EmbeddedTextRegion.from_coords(0, 0, 24, 24, text=text)
+    assert etr.extract_text(objects=None) == expected
+
+
 class MockDetectionModel(layout.UnstructuredObjectDetectionModel):
     def initialize(self, *args, **kwargs):
         pass