Function main

test/perf/compiled_extensions/command.py:57–227

()

def main():
    import argparse

    cases = Case._CASES

    parser = argparse.ArgumentParser(
        description="Compare implementation between them"
    )
    parser.add_argument(
        "case",
        help="Case to run",
        nargs="+",
        choices=["all"] + sorted(c.__name__ for c in cases),
    )
    parser.add_argument("--filter", help="filter the test for this regexp")
    parser.add_argument(
        "--factor", help="scale number passed to timeit", type=float, default=1
    )
    parser.add_argument("--csv", help="save to csv", action="store_true")
    save_group = parser.add_argument_group("Save result for later compare")
    save_group.add_argument(
        "--save-db",
        help="Name of the sqlite db file to use",
        const="perf.db",
        nargs="?",
    )
    save_group.add_argument(
        "--save-name",
        help="A name given to the current save. "
        "Can be used later to compare against this run.",
    )

    compare_group = parser.add_argument_group("Compare against stored data")
    compare_group.add_argument(
        "--compare-db",
        help="Name of the sqlite db file to read for the compare data",
        const="perf.db",
        nargs="?",
    )
    compare_group.add_argument(
        "--compare-filter",
        help="Filter the compare data using this string. Can include "
        "git-short-sha, save-name previously used or date. By default the "
        "latest values are used",
    )

    args = parser.parse_args()

    to_run: list[type[Case]]
    if "all" in args.case:
        to_run = cases
    else:
        to_run = [c for c in cases if c.__name__ in args.case]

    if args.save_db:
        save_engine = sa.create_engine(
            f"sqlite:///{args.save_db}", poolclass=sa.NullPool
        )