File size: 602 Bytes
d4c9875
 
 
 
 
 
 
 
 
 
 
 
8627bc6
d4c9875
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import pytest

from marker.renderers.markdown import MarkdownRenderer


@pytest.mark.config({"page_range": [0]})
@pytest.mark.filename("A17_FlightPlan.pdf")
def test_disable_extract_images(pdf_document):
    renderer = MarkdownRenderer({"extract_images": False})
    md = renderer(pdf_document).markdown

    # Verify markdown
    assert "jpeg" not in md


@pytest.mark.config({"page_range": [0]})
@pytest.mark.filename("A17_FlightPlan.pdf")
def test_extract_images(pdf_document):
    renderer = MarkdownRenderer()
    md = renderer(pdf_document).markdown

    # Verify markdown
    assert "jpeg" in md