annator / tests /smoke_test.py
techprotrade's picture
Add tests directory
eac0d05 verified
Raw
History Blame Contribute Delete
712 Bytes
import sys, pathlib
sys.path.insert(0, str(pathlib.Path(__file__).parent.parent))
from pathlib import Path
from src.core.models import JobConfig
from src.core.orchestrator import OfflinePDFOrchestrator
def test_init(tmp_path):
orc = OfflinePDFOrchestrator(work_dir=tmp_path)
cfg = JobConfig(input_pdf=tmp_path / "x.pdf", output_dir=tmp_path / "out")
assert cfg.job_id
assert cfg.pipeline == "full_document"
assert "full_document" in orc.list_available_pipelines()
assert "text_edit" in orc.list_available_pipelines()
print("Smoke test passed")
if __name__ == "__main__":
import tempfile
with tempfile.TemporaryDirectory() as t:
test_init(Path(t))