from base import BaseTest


def get_ret_val(res):
    assert len(res) == 1
    return res[0]


class TestNonReg1(BaseTest):

    def test_i_can_evaluate_python(self, sheerka, user):
        res = sheerka.evaluate_user_input("1 + 1", user)
        assert len(res) == 1
        ret_val = res[0]
        assert ret_val.status is True
        assert ret_val.value == 2

    def test_i_cannot_evaluate_variable_that_is_not_defined(self, sheerka, user):
        res = sheerka.evaluate_user_input("a", user)
        ret_val = get_ret_val(res)

        assert ret_val.status is False

    def test_i_can_remember_variables(self, sheerka, user):
        sheerka.evaluate_user_input("a = 10", user)

        res = sheerka.evaluate_user_input("a", user)
        ret_val = get_ret_val(res)

        assert ret_val.value == 10