From 67f3fcce99a753e92ab38b7b8d66e70b2518ce46 Mon Sep 17 00:00:00 2001
From: quyuan <quyuan@pjlab.org>
Date: Sat, 13 Jul 2024 17:29:45 +0800
Subject: [PATCH] add ci

---
 tests/test_cli/test_cli.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/tests/test_cli/test_cli.py b/tests/test_cli/test_cli.py
index 1402d283..1606195b 100644
--- a/tests/test_cli/test_cli.py
+++ b/tests/test_cli/test_cli.py
@@ -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)