This commit is contained in:
quyuan
2024-07-13 17:29:45 +08:00
parent 3a0039eb29
commit 67f3fcce99
+1 -5
View File
@@ -1,12 +1,8 @@
import pytest
import os
from conf import conf
import subprocess
from lib import common
import logging
import os
import json
from loguru import logger
from magic_pdf.pipe.UNIPipe import UNIPipe
from magic_pdf.rw.DiskReaderWriter import DiskReaderWriter
pdf_res_path = conf.conf["pdf_res_path"]
@@ -27,7 +23,7 @@ class TestCli:
demo_names.append(pdf_file.split('.')[0])
for demo_name in demo_names:
model_path = os.path.join(pdf_dev_path, f"{demo_name}_model.json")
pdf_path = os.path.join(pdf_dev_path, f"{demo_name}.pdf")
pdf_path = os.path.join(pdf_dev_path, "pdf", f"{demo_name}.pdf")
pdf_bytes = open(pdf_path, "rb").read()
model_json = json.loads(open(model_path, "r", encoding="utf-8").read())
image_writer = DiskReaderWriter(pdf_dev_path)