aboutsummaryrefslogtreecommitdiff
path: root/benchmark/agbenchmark/challenges/deprecated/code/3_modify/artifacts_out/test.py
blob: 5012e167ab63ea224de8094af638671f8e99b72b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from sample_code import multiply_int


def test_multiply_int(num: int, multiplier, expected_result: int) -> None:
    result = multiply_int(num, multiplier)
    print(result)
    assert (
        result == expected_result
    ), f"AssertionError: Expected the output to be {expected_result}"


if __name__ == "__main__":
    # test the trivial case
    num = 4
    multiplier = 2
    expected_result = 8
    test_multiply_int(num, multiplier, expected_result)

    # so its not hard coded
    num = 7
    multiplier = 7
    expected_result = 49
    test_multiply_int(num, multiplier, expected_result)

    # negative numbers
    num = -6
    multiplier = 2
    expected_result = -12
    test_multiply_int(num, multiplier, expected_result)