testsAndMisc/docs/DOCS-python-split-recipes.md
Krzysztof kuhy Rudnicki 49d82511c3 Adopt the four markdown namespaces and gate them
Renames this repo's markdown onto README.md / CLAUDE*.md / DOCS*.md /
TODO*.md so a future session can tell a live task from a finished
record without reading the file to find out.

TODO files carry the literal "REMOVE ME AFTER FINISH" and are deleted
when the work lands. Enforced by the md-naming pre-commit hook and CI
workflow, both delegating to the shared gate in kuhyx/utils so this
repo's idea of a valid name cannot drift from every other repo's.

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01A47aX5DJwmA3QFn3pkEWi7
2026-08-24 15:55:46 +02:00

9.9 KiB
Raw Blame History

Python split recipes for the 250-line cap

Settled empirically while clearing python_pkg/wsg_grabber. Every rule here cost at least one failed gate run or one revert to learn — do not re-derive them. Referenced from refactor_claude_todo_resume.md.

The re-export form

When a split moves a public name out of a module that callers/tests still reach through the old path, the re-export must be marked with __all__:

from python_pkg.pkg._new_module import show_logs, summarise

__all__ = ["build_parser", "main", ..., "show_logs", "summarise"]

The PEP 484 alias form from x import y as y does not work here: ruff raises PLC0414 (useless-import-alias), auto-fixes it away, and then removes the now-"unused" import under F401 — silently breaking cli.summarise. Verified on wsg_grabber/cli.py; __all__ passes ruff + mypy + pylint clean.

Splits that only move private helpers (leading _, no external caller) need no shim at all — check with grep first, as in catalog.py.

__all__ only needs the re-exported names, not every public name in the module — names defined locally stay reachable either way. Verified against the full gate on store.py.

__all__ does not save you from ruff's TC001. It suppresses F401, but a re-exported class used only in annotations gets rewritten into the TYPE_CHECKING block, which lints perfectly and raises AttributeError at import time. This nearly shipped a broken ui.Callbacks that cli.py constructs. A class that must be reachable at runtime from module X stays defined in module X. Always finish a re-export split with

PYTHONPATH=. python3 -c "from python_pkg.pkg import mod; mod.TheName"

mock.patch pins names to a module — run this grep FIRST

This is the constraint that decides where a seam can go. patch("mod.name") rebinds an attribute on mod. If you move a function to _new.py, that function resolves its collaborators through _new's globals, and the patch on mod no longer reaches it. Tests then fail with either <module> does not have the attribute 'x' or, worse, a real call to the thing that was supposed to be mocked.

Before choosing any seam, list every pinned name in the package:

grep -ohP 'patch\((?:f")?(?:\{MOD\}|"[\w.]+)\.\K[\w]+' <pkg>/tests/*.py | sort -u

Everything it prints — including plain imports like subprocess, shutil, Path, time, and collaborators imported from sibling modules — must stay resolvable in the module the test names. Two consequences:

  • A function that calls a patched collaborator has to stay in that module, or move together with it.
  • patch.object(mod, "tk", fake) is the same rule in another spelling. test_ui.py uses it, which is why every tkinter line had to stay in ui.py — moving widget construction out errored 22 tests.

brother_printer is the extreme case: its test modules patch ~90 distinct names across the package, including estimate_consumable_life on display. An otherwise clean extraction of the page-count block was reverted because of exactly that. Grep first, then pick the seam.

Tests that call private methods pin them too — test_player.py calls mpv._note_errors(...), so that method stayed as a one-line delegate to the moved implementation. That is the escape hatch for a one-off pinned name: leave a thin delegate behind. It does not scale — under select = ["ALL"] every delegate needs a full docstring with Args:, so a 25-line function leaves a 12-line stub. Eight of those to make a line count is the cap-gaming the spec forbids.

A mispointed patch can HANG rather than fail

In brother_printer a stale patch target raised AttributeError or made a real subprocess call — loud, or at worst a side effect you could check for afterwards. In code_tutor it spins: _write_tests_first_flow reads from an input_fn in a retry loop, so an unintercepted _collect_and_rate_tests ran the real interactive flow and the suite hung for two minutes with no output. ps showed it State: R, not blocked on I/O.

Two consequences for any package whose code prompts, retries or shells out:

  • Run the suite under timeout. A hang is indistinguishable from a slow run until you have waited longer than you should.
  • Assert the patch targets resolve before invoking pytest, since a missing one is exactly what causes the hang:
PYTHONPATH=. python3 -c "
from python_pkg.pkg import mod_a as a, mod_b as b
for n in ('_helper', 'Thing'): assert hasattr(a, n), n
"

code_tutor also patches by absolute string with no MOD constant, so a seam there costs one rewrite per patch string rather than one sed. Prefer seams whose moved block has few patched names, even at slightly worse line arithmetic. Watch out for a sed with escaped alternation silently matching nothing — a no-op rewrite looks exactly like a successful one at the shell, so grep the result before trusting it.

You may edit the tests to follow the code (agreed 2026-08-16)

When a patch target moves, update the patch target. Bounded by:

  • no assertion changed,
  • no test deleted, skipped or weakened,
  • the pass count identical before and after (record it in the evidence).

Updating MOD = "python_pkg.pkg.display" to the function's new home is bookkeeping, not weakening. Do it per patched name, never as a blanket swap: a name patched from two different test modules (_display_report_header is patched by both test_display.py and test_display_part2.py) has to stay reachable from both, so it belongs in whichever module keeps the shared entry points.

The test modules are usually the best map of where the seam goes. In brother_printer, test_display.py patches only USB-side helpers, test_display_part2.py only network-side, and test_display_part3.py only the page-count collaborators — three concerns in one 427-line module. Split the source to match, and let each test module's MOD follow.

Budget the new import block and __all__ before deciding a two-way split is enough. Five splits so far landed 2–16 lines over the cap after the first cut and needed a further seam; the header costs 10–20 lines that the arithmetic on function sizes does not show.

Splitting a class: if tests call private methods as bound attributes (worker._fetch_thread(...) — grep the tests first), module-level functions will not do; use mixins. The shape that passes every gate with no suppression (settled on downloader.py, do not re-derive):

# _worker_base.py
class WorkerBase(ABC):
    _deps: WorkerDeps          # bare annotations for read-only attributes
    _downloaded: int = 0       # real default for any attribute the mixin ASSIGNS

    @abstractmethod
    def _publish(self, event: DownloadEvent) -> None:
        """Supplied by the concrete Worker."""

Both mixins inherit WorkerBase; Worker(ScanMixin, DownloadMixin). Why not the obvious alternatives:

  • Naming a class *Mixin does not get you pylint's no-member exemption here — ignored-checks-for-mixins is not configured for it. Don't rely on it.
  • raise NotImplementedError stub bodies are uncovered lines under the 100% branch gate.
  • if TYPE_CHECKING: def _publish(...) -> None: ... satisfies mypy but leaves pylint reporting no-member.
  • A bare annotation (_downloaded: int) is not enough for an attribute the mixin assigns to (+= 1); pylint wants a real default.

Watch the MRO: Worker(ScanMixin, DownloadMixin) means ScanMixin wins any name collision.

Coverage: meta/pyproject.toml sets source = ["python_pkg"], so a new sibling module is measured automatically. Confirmed: new modules show up in the report at 100% rather than being silently skipped. Add no new branches (no if TYPE_CHECKING guards beyond what moved, no defensive try/except).

The seam-selection rule (settled over 8 packages)

Generalised from python_pkg; full detail in docs/python-split-recipes.md.

A block may move if every MOD-relative patch that reaches it comes from one test module. If two test modules patch a shared collaborator, keep the collaborator and move its callers instead. The caller resolves the name through its own module's globals, so importing it into the new module makes the patch land there.

Corollaries that each cost a failed run to learn:

  • test_X.py needing no edit is the signal you cut in the right place. It held on four of five source splits in brother_printer.
  • Never blanket-sed a MOD. Decide per test by which function is under test. test_challenge_part3.py needed 13 of its strings repointed and 4 left alone.
  • A mispointed patch does not always fail — it can HANG or do real work. code_tutor spun for two minutes running a real interactive flow; cups_queue would have restarted the live CUPS daemon. Always run under timeout, and assert the targets resolve before invoking pytest.
  • When splitting a test file, copy the WHOLE preamble (docstring, imports, fixture builders) into the new file and let ruff's F401 pass prune it. Hand-writing the import block failed on every early attempt; the copy-then- prune order removed 82 imports correctly in wsg_grabber with none missed. It does not cover module-level defs and constants sitting between sections — _make_verifier, _QUESTIONS, _logger and _SAME_FIELD_X_TOLERANCE_PX were each left behind that way. Grep for them.
  • Budget two cuts. Nine of the splits landed 1–20 lines over after the first seam. Pure-data blocks (prompt strings, skip tables, SVG fragments) are the cheapest second cut and have no patch targets.
  • Prefer a conftest.py fixture over a duplicated builder — that is where _make_verifier went, and it removed a shadowing copy of _make_live_mock.

Two helper scripts are in the scratchpad, not committed (they are throwaway): they cut a module at a banner comment or at a named def/class and copy the preamble. Rewriting them takes about five minutes.