fixed #18 : I can evaluate concept

tests/non_reg/test_non_reg1.py

@@ -1,6 +1,11 @@
 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):
@@ -10,9 +15,16 @@ class TestNonReg1(BaseTest):
         assert ret_val.status is True
         assert ret_val.value == 2
 
-    def test_i_can_evaluate_variable_that_is_not_defined(self, sheerka, user):
+    def test_i_cannot_evaluate_variable_that_is_not_defined(self, sheerka, user):
         res = sheerka.evaluate_user_input("a", user)
-        assert len(res) == 1
-        ret_val = res[0]
+        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