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)