Modules and imports

# helpers.py, next to this file, defines average(scores) and grade(score).
# Import them, then print the average and the grade of the scores typed in.
scores = [int(word) for word in input().split()]
print("average:", 0)

from learnrun import run_main


def report(line):
    return run_main(line + "\n").strip().splitlines()


def test_average_and_grade():
    """80 65 gives average 72.5 and grade C"""
    got = report("80 65")
    assert got == ["average: 72.5", "grade: C"], f"for 80 65 the program printed {got!r}"


def test_top_grade():
    """90 100 gives average 95.0 and grade A"""
    got = report("90 100")
    assert got == ["average: 95.0", "grade: A"], f"for 90 100 the program printed {got!r}"


def test_uses_helpers():
    """main.py imports helpers instead of copying its functions"""
    with open("main.py", encoding="utf-8") as f:
        source = f.read()
    assert "import helpers" in source, "main.py has no line import helpers"
    assert "from helpers import grade" in source, "main.py has no line from helpers import grade"
    assert "def " not in source, "main.py defines its own function; use the ones in helpers.py"

"""Score helpers for main.py. Leave this file as it is."""


def average(scores: list[int]) -> float:
    return sum(scores) / len(scores)


def grade(score: float) -> str:
    if score >= 90:
        return "A"
    if score >= 75:
        return "B"
    if score >= 50:
        return "C"
    return "F"

from helpers import shout


def greet(name: str) -> str:
    return shout("hello, " + name)


name = input()
print(greet(name))

import contextlib
import importlib
import io
import sys

from learnrun import run_main


def import_main():
    sys.modules.pop("main", None)
    printed = io.StringIO()
    saved = sys.stdin
    sys.stdin = io.StringIO("")
    try:
        with contextlib.redirect_stdout(printed):
            module = importlib.import_module("main")
    finally:
        sys.stdin = saved
    return module, printed.getvalue()


def test_import_is_quiet():
    """Importing main.py reads no input and prints nothing"""
    try:
        _, printed = import_main()
    except EOFError:
        assert False, "importing main.py called input(): move it under if __name__ == \"__main__\":"
    assert printed == "", f"importing main.py printed {printed!r}: move the print under if __name__ == \"__main__\":"


def test_greet():
    """greet(\"ada\") returns HELLO, ADA!"""
    try:
        module, _ = import_main()
    except EOFError:
        assert False, "main.py could not be imported: it called input() at the top level"
    got = module.greet("ada")
    assert got == "HELLO, ADA!", f"greet(\"ada\") returned {got!r}, expected 'HELLO, ADA!'"


def test_run_as_script():
    """Run as a script, it still asks for a name and greets it"""
    got = run_main("Grace\n").strip()
    assert got == "HELLO, GRACE!", f"for Grace the program printed {got!r}, expected 'HELLO, GRACE!'"

"""Text helpers for main.py. Leave this file as it is."""


def shout(text: str) -> str:
    return text.upper() + "!"