File size: 1,006 Bytes
66acddd 18eefbb 66acddd 18eefbb 66acddd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import pytest
from marker.converters.table import TableConverter
from marker.renderers.markdown import MarkdownOutput
from marker.util import classes_to_strings
def _table_converter(config, model_dict, renderer, temp_pdf):
converter = TableConverter(
artifact_dict=model_dict,
processor_list=None,
renderer=classes_to_strings([renderer])[0],
config=config
)
markdown_output: MarkdownOutput = converter(temp_pdf.name)
markdown = markdown_output.markdown
assert len(markdown) > 0
assert "cyclic" in markdown
@pytest.mark.output_format("markdown")
@pytest.mark.config({"page_range": [5]})
def test_table_converter(config, model_dict, renderer, temp_doc):
_table_converter(config, model_dict, renderer, temp_doc)
@pytest.mark.output_format("markdown")
@pytest.mark.config({"page_range": [5], "force_ocr": True})
def test_table_converter_ocr(config, model_dict, renderer, temp_doc):
_table_converter(config, model_dict, renderer, temp_doc)
|