Add test

VikParuchuri · Jan 24, 2025 · 5fc0546 · 5fc0546
1 parent d3d08c3
commit 5fc0546
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/processors/test_llm_processors.py b/tests/processors/test_llm_processors.py
@@ -65,7 +65,7 @@ def test_llm_table_processor(pdf_document, detection_model, table_rec_model, rec
         <td>Column 4</td>
     </tr>
     <tr>
-        <td>Value 1</td>
+        <td>Value 1 <math>x</math></td>
         <td>Value 2</td>
         <td>Value 3</td>
         <td>Value 4</td>
@@ -93,6 +93,9 @@ def test_llm_table_processor(pdf_document, detection_model, table_rec_model, rec
     table_cells = tables[0].contained_blocks(pdf_document, (BlockTypes.TableCell,))
     assert table_cells[0].text == "Column 1"
 
+    markdown = MarkdownRenderer()(pdf_document).markdown
+    assert "Value 1 $x$" in markdown
+
 
 @pytest.mark.filename("adversarial.pdf")
 @pytest.mark.config({"page_range": [0]})