import pytest

from phasm.type5.solver import Type5SolverException

from ..helpers import Suite


@pytest.mark.integration_test
def test_expr_constant_literal_does_not_fit_module_constant():
    code_py = """
CONSTANT: u8 = 1000

@exported
def testEntry() -> u8:
    return CONSTANT
"""

    with pytest.raises(Type5SolverException, match=r'Must fit in 1 byte\(s\)'):
        Suite(code_py).run_code()

@pytest.mark.integration_test
def test_expr_constant_literal_does_not_fit_return():
    code_py = """
@exported
def testEntry() -> u8:
    return 1000
"""

    with pytest.raises(Type5SolverException, match=r'Must fit in 1 byte\(s\)'):
        Suite(code_py).run_code()