Selaa lähdekoodia

Merge pull request #2256 from myhloli/dev

fix(test_table): update image path to use relative path
Xiaomeng Zhao 7 kuukautta sitten
vanhempi
commit
4bd3381c92
1 muutettua tiedostoa jossa 2 lisäystä ja 1 poistoa
  1. 2 1
      tests/unittest/test_table/test_rapidtable.py

+ 2 - 1
tests/unittest/test_table/test_rapidtable.py

@@ -1,4 +1,5 @@
 import unittest
+import os
 from PIL import Image
 from lxml import etree
 
@@ -8,7 +9,7 @@ from magic_pdf.model.sub_modules.table.rapidtable.rapid_table import RapidTableM
 
 class TestppTableModel(unittest.TestCase):
     def test_image2html(self):
-        img = Image.open("assets/table.jpg")
+        img = Image.open(os.path.join(os.path.dirname(__file__), "assets/table.jpg"))
         atom_model_manager = AtomModelSingleton()
         ocr_engine = atom_model_manager.get_atom_model(
             atom_model_name='ocr',