test_main.py

import pytest
from typer.testing import CliRunner
from main import app

runner = CliRunner()


@pytest.mark.parametrize("n", [1, 2])
def test_parser(n: int):
    result = runner.invoke(app, ["parse", f"tests/parser/input_{n}.txt"])
    assert result.exit_code == 0
    assert result.stdout == ""


@pytest.mark.parametrize(
    "n,line_numbers",
    [
        (1, [1, 3, 7]),
        (2, [2]),
        (3, [1, 4]),
    ],
)
def test_parser_errors(n: int, line_numbers: list[int]):
    result = runner.invoke(app, ["parse", f"tests/parser/invalid_input_{n}.txt"])
    assert result.exit_code == 0
    for ln in line_numbers:
        assert f"line {ln}" in result.stdout
    assert result.stdout.lower().count("line") == len(line_numbers)


@pytest.mark.parametrize("n", [1, 2, 3])
def test_ast(n: int):
    result = runner.invoke(app, ["ast", f"tests/ast/input_{n}.txt"])
    assert result.exit_code == 0
    with open(f"tests/ast/output_{n}.txt", encoding="utf-8") as f:
        output = f.read()
    assert result.stdout == output


@pytest.mark.parametrize(
    "name,line_numbers,additional",
    [
        ("break", [1], "break"),
        ("continue", [1], "continue"),
        ("vector", [1, 3, 7], "vector"),
        ("variables", [5, 7], "variable"),
        ("transpose", [7], "transpose"),
        ("special_matrix", [1, 11], "matrix"),
        ("indexing", [5, 6, 7], ""),
        ("indexing_bounds", [4, 11, 12], "out of bounds"),
        ("binary_operations", [7, 8, 14, 16, 17], "binary operation"),
        ("comparisons", [7, 9], "comparison"),
        ("compound_assignments", [10, 11, 12], "assignment"),
        ("ranges", [9], "range"),
        ("redeclaration", [12], ""),
    ],
)
def test_sem_errors(name: str, line_numbers: list[int], additional: str):
    result = runner.invoke(app, ["sem", f"tests/semantic/{name}.txt"])
    assert result.exit_code == 0
    for ln in line_numbers:
        assert f"line {ln}" in result.stdout
    assert result.stdout.lower().count("line") == len(line_numbers)
    if additional:
        assert result.stdout.lower().count(additional) == len(line_numbers)


@pytest.mark.parametrize(
    "name,output",
    [
        ("simple_math", [23, 35, 1.0, 1.0, 1, -2, 9.0, 6.0, 1.0]),
        ("strings", ["aaa", "abcd"]),
        ("conditions", [0, 1, 0, 1, 0, 1, 0, 1]),
        (
            "vectors",
            [
                [1, 2, 3],
                [[1, 0, 0], [0, 1, 0], [0, 0, 1]],
                [0, 0],
                [[1, 1]],
                [[[0], [0]], [[0], [0]], [[0], [0]]],
                [[1, 1, 1]],
            ],
        ),
        ("variables", [2, 1, 3, "OK", 6]),
        ("while", [4, 3, 2, 1, 0]),
        ("for", [1, 10, 2, 10, 3, 10, 4, 10]),
        ("break_continue", [1, 2, 1, 2, 4] * 2),
        (
            "element_reference",
            [
                [1, 0],
                0,
                [[1, 2], [0, 1]],
                [[0, 2], [0, 1]],
                [[0, 0], [0, 1]],
            ],
        ),
        (
            "mat_operators",
            [
                [[2, 2], [2, 2]],
                [[4, 4], [4, 4]],
                [[3, 3], [3, 3]],
            ],
        ),
    ],
)
def test_interpreter(name: str, output: str):
    result = runner.invoke(app, ["run", f"tests/interpreter/{name}.txt"])
    assert result.exit_code == 0
    assert result.stdout == "\n".join(map(str, output)) + "\n"


def test_interpreter_return():
    result = runner.invoke(app, ["run", "tests/interpreter/return.txt"])
    assert result.exit_code == 1


@pytest.mark.parametrize(
    "name",
    [
        "parser/input_1.txt",
        "parser/input_2.txt",
        "ast/input_1.txt",
        "ast/input_2.txt",
        "ast/input_3.txt",
    ],
)
def test_all_on_previous_tests(
    name: str,
):  # run interpreter on parser tests and AST tests
    result = runner.invoke(app, ["run", f"tests/{name}"])
    assert result.exit_code == 0