Kommandozeilenprogramme mit argparse und sys

import argparse
import sys


def main(argv: list[str] | None = None) -> int:
    parser = argparse.ArgumentParser(prog="greet")
    parser.add_argument("name")
    # Add --times (an int, default 1) and the flag --shout.
    args = parser.parse_args(argv)
    print(f"Hello, {args.name}!")
    return 0


if __name__ == "__main__":
    sys.exit(main())

import contextlib
import io

from main import main


def run(argv):
    out = io.StringIO()
    err = io.StringIO()
    with contextlib.redirect_stdout(out), contextlib.redirect_stderr(err):
        try:
            status = main(argv)
        except SystemExit as stop:
            status = stop.code
    return status, out.getvalue().splitlines(), err.getvalue()


def test_default():
    """Ada allein ergibt eine Begrüßung und Status 0"""
    status, lines, err = run(["Ada"])
    assert (status, lines) == (0, ["Hello, Ada!"]), f"main(['Ada']) gab {status!r} zurück und {lines!r} aus; stderr: {err!r}"


def test_times():
    """--times 3 gibt die Begrüßung dreimal aus"""
    status, lines, err = run(["Ada", "--times", "3"])
    assert lines == ["Hello, Ada!"] * 3, f"mit --times 3 gab das Programm {lines!r} aus; stderr: {err!r}"


def test_shout():
    """--shout schreibt groß, vor oder nach dem Namen"""
    status, lines, err = run(["--shout", "Bo"])
    assert lines == ["HELLO, BO!"], f"mit --shout gab das Programm {lines!r} aus; stderr: {err!r}"


def test_bad_times():
    """--times many endet mit Exit-Status 2"""
    status, lines, err = run(["Ada", "--times", "many"])
    assert status == 2, f"mit --times many war der Status {status!r}, erwartet 2 (lassen Sie type=int ablehnen)"

import argparse
import sys


def main(argv: list[str] | None = None) -> int:
    parser = argparse.ArgumentParser(prog="average")
    parser.add_argument("numbers", nargs="+", type=float)
    parser.add_argument("--max", type=float, default=100.0)
    args = parser.parse_args(argv)
    for number in args.numbers:
        if number > args.max:
            print(f"average: {number} is above the maximum {args.max}")
    print(f"{sum(args.numbers) / len(args.numbers):.2f}")
    return 0


if __name__ == "__main__":
    main()

import contextlib
import io
import sys

from learnrun import run_main
from main import main


def run(argv):
    out = io.StringIO()
    err = io.StringIO()
    with contextlib.redirect_stdout(out), contextlib.redirect_stderr(err):
        try:
            status = main(argv)
        except SystemExit as stop:
            status = stop.code
    return status, out.getvalue(), err.getvalue()


def test_average():
    """2, 4 und 9 ergeben 5.00 und Status 0"""
    status, out, err = run(["2", "4", "9"])
    assert (status, out.strip()) == (0, "5.00"), f"main gab {status!r} zurück und {out!r} aus"


def test_above_max():
    """500 wird auf stderr gemeldet, mit Status 1 und leerem stdout"""
    status, out, err = run(["5", "500"])
    assert status == 1, f"main gab {status!r} zurück, erwartet 1"
    assert out == "", f"stdout sollte leer sein, enthält aber {out!r}"
    assert err.strip() == "average: 500.0 is above the maximum 100.0", f"stderr enthält {err!r}"


def test_custom_max():
    """--max 10 weist 11 ab"""
    status, out, err = run(["--max", "10", "5", "11"])
    assert status == 1, f"mit --max 10 und 11 gab main {status!r} zurück, erwartet 1"


def test_not_a_number():
    """Ein Wort statt einer Zahl endet mit Status 2"""
    status, out, err = run(["five"])
    assert status == 2, f"für five war der Status {status!r}, erwartet 2"


def test_status_reaches_the_shell():
    """Als Programm endet python main.py 5 500 mit Status 1"""
    saved = sys.argv
    sys.argv = ["main.py", "5", "500"]
    code = 0
    try:
        with contextlib.redirect_stderr(io.StringIO()):
            run_main()
    except SystemExit as stop:
        code = stop.code
    finally:
        sys.argv = saved
    assert code == 1, f"das Programm endete mit Status {code!r}, erwartet 1: rufen Sie sys.exit(main()) auf"