mirror of
https://github.com/docling-project/docling.git
synced 2025-06-27 05:20:05 +00:00
30 lines
756 B
Python
30 lines
756 B
Python
import os
|
|
|
|
|
|
def _setup_env():
|
|
os.environ["DOCLING_PERF_PAGE_BATCH_SIZE"] = "12"
|
|
os.environ["DOCLING_DEBUG_VISUALIZE_RAW_LAYOUT"] = "True"
|
|
os.environ["DOCLING_ARTIFACTS_PATH"] = "/path/to/artifacts"
|
|
|
|
|
|
def test_settings():
|
|
_setup_env()
|
|
|
|
import importlib
|
|
|
|
import docling.datamodel.settings as m
|
|
|
|
# Reinitialize settings module
|
|
importlib.reload(m)
|
|
|
|
# Check top level setting
|
|
assert str(m.settings.artifacts_path) == "/path/to/artifacts"
|
|
|
|
# Check nested set via environment variables
|
|
assert m.settings.perf.page_batch_size == 12
|
|
assert m.settings.debug.visualize_raw_layout is True
|
|
|
|
# Check nested defaults
|
|
assert m.settings.perf.doc_batch_size == 2
|
|
assert m.settings.debug.visualize_ocr is False
|