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.
14 lines
396 B
Python
14 lines
396 B
Python
from PIL import Image, ImageDraw
|
|
|
|
|
|
def test_latex_ocr(texify_predictor):
|
|
img = Image.new('RGB', (200, 100), color='white')
|
|
draw = ImageDraw.Draw(img)
|
|
draw.text((10, 10), "E = mc2", fill='black', font_size=48)
|
|
|
|
results = texify_predictor([img])
|
|
text = results[0].text.strip()
|
|
assert len(results) == 1
|
|
|
|
assert text.startswith("<math")
|
|
assert text.endswith("</math>") |