mirror of
https://github.com/Tencent/WeKnora.git
synced 2026-06-04 13:30:32 +08:00
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
import os
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from docreader import config
|
|
|
|
|
|
class DocReaderConfigTest(unittest.TestCase):
|
|
def test_parser_concurrency_defaults_are_conservative(self):
|
|
with patch.dict(os.environ, {}, clear=True):
|
|
cfg = config.load_config()
|
|
|
|
self.assertEqual(cfg.markitdown_max_workers, 1)
|
|
self.assertEqual(cfg.pdf_render_max_workers, 1)
|
|
self.assertEqual(cfg.pdf_render_dpi, 200)
|
|
self.assertEqual(cfg.pdf_jpeg_quality, 90)
|
|
|
|
def test_loads_parser_concurrency_env(self):
|
|
env = {
|
|
"DOCREADER_MARKITDOWN_MAX_WORKERS": "3",
|
|
"DOCREADER_PDF_RENDER_MAX_WORKERS": "2",
|
|
"DOCREADER_PDF_RENDER_DPI": "180",
|
|
"DOCREADER_PDF_JPEG_QUALITY": "85",
|
|
}
|
|
with patch.dict(os.environ, env):
|
|
cfg = config.load_config()
|
|
|
|
self.assertEqual(cfg.markitdown_max_workers, 3)
|
|
self.assertEqual(cfg.pdf_render_max_workers, 2)
|
|
self.assertEqual(cfg.pdf_render_dpi, 180)
|
|
self.assertEqual(cfg.pdf_jpeg_quality, 85)
|
|
|
|
def test_dump_config_includes_parser_limits(self):
|
|
dumped = config.dump_config()
|
|
|
|
self.assertIn("DOCREADER_MARKITDOWN_MAX_WORKERS", dumped)
|
|
self.assertIn("DOCREADER_PDF_RENDER_MAX_WORKERS", dumped)
|
|
self.assertIn("DOCREADER_PDF_RENDER_DPI", dumped)
|
|
self.assertIn("DOCREADER_PDF_JPEG_QUALITY", dumped)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|