You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
562 B
Python
18 lines
562 B
Python
1 month ago
|
from marker.providers.image import ImageProvider
|
||
|
|
from marker.renderers.markdown import MarkdownOutput
|
||
|
|
|
||
|
|
|
||
|
|
def test_image_provider(config, temp_image):
|
||
|
|
provider = ImageProvider(temp_image.name, config)
|
||
|
|
assert len(provider) == 1
|
||
|
|
assert provider.get_images([0], 72)[0].size == (512, 512)
|
||
|
|
|
||
|
|
page_lines = provider.get_page_lines(0)
|
||
|
|
assert len(page_lines) == 0
|
||
|
|
|
||
|
|
def test_image_provider_conversion(pdf_converter, temp_image):
|
||
|
|
markdown_output: MarkdownOutput = pdf_converter(temp_image.name)
|
||
|
|
assert "Hello, World!" in markdown_output.markdown
|
||
|
|
|
||
|
|
|