format

Author: qued

test_unstructured/partition/pdf_image/test_merge_elements.py

@@ -14,15 +14,11 @@ def test_text_source_preserved_during_merge():
 
     # Create two simple LayoutElements with different text_source values
     inferred_element = LayoutElement(
-        bbox=Rectangle(0, 0, 100, 50),
-        text=None,
-        is_extracted=IsExtracted.FALSE
+        bbox=Rectangle(0, 0, 100, 50), text=None, is_extracted=IsExtracted.FALSE
     )
 
     extracted_element = LayoutElement(
-        bbox=Rectangle(0, 0, 100, 50),
-        text="Extracted text",
-        is_extracted=IsExtracted.TRUE
+        bbox=Rectangle(0, 0, 100, 50), text="Extracted text", is_extracted=IsExtracted.TRUE
     )
 
     # Create LayoutElements arrays
@@ -48,5 +44,5 @@ def test_text_source_preserved_during_merge():
     # Check the merged page's elements_array
     merged_page = merged_layout.pages[0]
     assert "Extracted text" in merged_page.elements_array.texts
-    assert hasattr(merged_page.elements_array, 'is_extracted_array')
+    assert hasattr(merged_page.elements_array, "is_extracted_array")
     assert IsExtracted.TRUE in merged_page.elements_array.is_extracted_array