diff --git a/src/ocrmypdf/__init__.py b/src/ocrmypdf/__init__.py index bc2e6791..fd06f2be 100644 --- a/src/ocrmypdf/__init__.py +++ b/src/ocrmypdf/__init__.py @@ -44,4 +44,4 @@ from . import hocrtransform from . import leptonica from . import pdfa from . import pdfinfo -from .run import run_pipeline as run +from . import run diff --git a/tests/test_unpaper.py b/tests/test_unpaper.py index 23f6d698..9d5b60df 100644 --- a/tests/test_unpaper.py +++ b/tests/test_unpaper.py @@ -22,7 +22,8 @@ from unittest.mock import MagicMock, patch import pytest -from ocrmypdf import __main__ as main +from ocrmypdf.__main__ import parser +from ocrmypdf.run import check_options from ocrmypdf.exceptions import ExitCode from ocrmypdf.exec import unpaper @@ -52,12 +53,12 @@ def spoof_unpaper_oldversion(tmpdir_factory): def test_no_unpaper(resources, no_outpdf): input_ = fspath(resources / "c02-22.pdf") output = fspath(no_outpdf) - options = main.parser.parse_args(args=["--clean", input_, output]) + options = parser.parse_args(args=["--clean", input_, output]) with patch("ocrmypdf.exec.unpaper.version") as mock_unpaper_version: mock_unpaper_version.side_effect = FileNotFoundError("unpaper") with pytest.raises(SystemExit): - main.check_options(options, log=MagicMock()) + check_options(options, log=MagicMock()) def test_old_unpaper(spoof_unpaper_oldversion, resources, no_outpdf):