fix: formatting errors

navapbc · Dec 18, 2024 · da6ddfc · da6ddfc
1 parent 7c2e809
commit da6ddfc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/app/tests/src/ingestion/test_pdf_stylings.py b/app/tests/src/ingestion/test_pdf_stylings.py
@@ -8,7 +8,7 @@ def test_styling_dataclass():
         pageno=1,
         headings=[Heading(title="Test Heading", level=1, pageno=1)],
         wider_text="test text in context",
-        bold=True
+        bold=True,
     )
     assert styling.text == "test text"
     assert styling.pageno == 1

diff --git a/app/tests/src/test_format.py b/app/tests/src/test_format.py
@@ -8,9 +8,9 @@
     FormattingConfig,
     _format_guru_to_accordion_html,
     _get_breadcrumb_html,
+    build_accordions,
     format_guru_cards,
     reify_citations,
-    build_accordions,
 )
 from src.retrieve import retrieve_with_scores
 from tests.src.db.models.factories import ChunkFactory
@@ -103,16 +103,13 @@ def test_reify_citations():
     subsections = split_into_subsections(chunks, factory=CitationFactory())
     config = FormattingConfig()
 
-    assert (
-        reify_citations("This is a citation (citation-0)", [], config, None)
-        == "This is a citation "
-    )
+    assert reify_citations("This is a citation (citation-0)", [], config, None) == "This is a citation "
 
     result = reify_citations(
         f"This is a citation ({subsections[0].id}) and another ({subsections[1].id}).",
         subsections,
         config,
-        None
+        None,
     )
 
     # Check that citations were added