from .helpers import Suite

def test_return():
    code_py = """
@exported
def testEntry() -> i32:
    return 13
"""

    result = Suite(code_py, 'test_fib').run_code()

    assert 13 == result.returned_value
    assert [] == result.log_int32_list

def test_addition():
    code_py = """
@exported
def testEntry() -> i32:
    return 10 + 3
"""

    result = Suite(code_py, 'test_fib').run_code()

    assert 13 == result.returned_value
    assert [] == result.log_int32_list

def test_call():
    code_py = """
def helper(left: i32, right: i32) -> i32:
    return left + right

@exported
def testEntry() -> i32:
    return helper(10, 3)
"""

    result = Suite(code_py, 'test_fib').run_code()

    assert 13 == result.returned_value
    assert [] == result.log_int32_list