orchestrata_tools.test_generator.testgen

Reference › orchestrata_tools › test_generator › testgen › init

The package entry point re-exports the public names of branch, config, generator, ranking and structural, and is imported by the script scripts.serialize_codebase_ast (see tests/scripts/test_ast_branches.py).

At a glance

Class diagram

No classes; see Functions.

Dependency diagram

API

No public API: the module defines no classes, functions or constants.

Source

Permalink: https://gitlab.com/MSamy/orchestrata-ultimate-architecture/-/blob/a4cbc2bdf/src/orchestrata_tools/test_generator/testgen/__init__.py

"""orchestrata_tools.test_generator.testgen — Layer 1 + 2 test generator.

Restored 2026-08-19 from ``scripts/ast_serializer/testgen`` (deleted
2026-07-09, commit 6a186c9d as part of removing a duplicate AST-parsing
layer). ``testgen`` had no replacement — ``scripts/serialize_codebase_ast.py
gen-tests`` emits a structurally different scaffold (no per-branch true/
false condition-derived inputs, no dataclass/enum/abstract/protocol
structural tests) — so this restores it, re-pointed at the CURRENT AST
layer (``orchestrata_ast`` + ``scripts/serialize_codebase_ast.py``) instead
of the deleted parser. See ``adapter.py`` for the bridge and
``scripts/serialize_codebase_ast.py``'s ``testgen`` subcommand for the CLI
entry point.

Public API
----------
    generate_tests(codebase, trees, out_dir, ...)  -> dict[path, source]
    print_ranking(codebase, top=30)                -> None (stdout)

    # Fine-grained
    rank_functions(codebase, ...)               -> list[RankedEntry]
    gen_dataclass_tests(name, extra, mod)       -> str
    gen_enum_tests(name, extra, mod)            -> str
    gen_abstract_tests(name, extra, mod)        -> str
    gen_protocol_tests(name, extra, mod)        -> str
    gen_branch_tests(...)                       -> str

    # Configuration
    ProjectConfig                               -> dataclass
    ORCHESTRATA_ULTIMATE                        -> pre-defined config
"""

from orchestrata_tools.test_generator.testgen.branch import gen_branch_tests
from orchestrata_tools.test_generator.testgen.config import (
    ORCHESTRATA_ULTIMATE,
    ProjectConfig,
    get_config,
)
from orchestrata_tools.test_generator.testgen.generator import (
    generate_tests,
    print_ranking,
)
from orchestrata_tools.test_generator.testgen.ranking import (
    RankedEntry,
    rank_functions,
    summarise,
)
from orchestrata_tools.test_generator.testgen.structural import (
    gen_abstract_tests,
    gen_dataclass_tests,
    gen_entry_point_tests,
    gen_enum_tests,
    gen_protocol_tests,
)

__all__ = [
    "generate_tests",
    "print_ranking",
    "rank_functions",
    "RankedEntry",
    "summarise",
    "gen_dataclass_tests",
    "gen_enum_tests",
    "gen_abstract_tests",
    "gen_protocol_tests",
    "gen_entry_point_tests",
    "gen_branch_tests",
    "ProjectConfig",
    "ORCHESTRATA_ULTIMATE",
    "get_config",
]