allow print call inside qlassf functions

TODO.md

@@ -43,14 +43,14 @@
 - [x] Allow constant functions
 - [x] Doc: emphatize the compiler flow
 - [x] Doc: properly render documentation
+- [x] Builtin debug functions: print()
 
 ### Week 4: (16 Oct 23)
 - [ ] Publish doc
 - [ ] Fix structure and typing location
 - [ ] Int arithmetic expressions (+, -, *, /)
 - [ ] Function call
 - [ ] Builtin functions: sum(), max(), min(), len()
-- [ ] Builtin debug functions: print()
 
 ## Month 2: 
 

qlasskit/ast2logic/t_ast.py

@@ -43,6 +43,10 @@ def translate_ast(fun) -> LogicFun:
     exps = []
     for stmt in fun.body:
         s_exps, env = translate_statement(stmt, env)
+
+        if s_exps == None:
+            continue 
+
         exps.append(s_exps)
 
     exps_flat = flatten(exps)

qlasskit/ast2logic/t_expression.py

@@ -220,16 +220,26 @@ def unfold(v_exps, op):
         else:
             raise exceptions.ExpressionNotHandledException(expr)
 
+    # Binop
     elif isinstance(expr, ast.BinOp):
         # Add | Sub | Mult | MatMult | Div | Mod | Pow | LShift | RShift
         # | BitOr | BitXor | BitAnd | FloorDiv
         raise exceptions.ExpressionNotHandledException(expr)
 
+    # Call
+    elif isinstance(expr, ast.Call):
+        if expr.func.id == 'print':
+            return (None, None)
+
+        else:
+            raise exceptions.ExpressionNotHandledException(expr)
+
     # Lambda
     # Dict
     # Set
     # Call
     # List
+
 
     else:
         raise exceptions.ExpressionNotHandledException(expr)

qlasskit/ast2logic/t_statement.py

@@ -79,5 +79,9 @@ def translate_statement(  # noqa: C901
         # Continue
         # Match
 
+    elif isinstance(stmt, ast.Expr):
+        texp, vexp = translate_expression(stmt.value, env)
+        return None, env
+
     else:
         raise exceptions.StatementNotHandledException(stmt)

test/test_qlassf.py

@@ -17,7 +17,14 @@
 from qlasskit import Qint4, QlassF, qlassf
 
 from . import utils
-
+from .utils import COMPILATION_ENABLED, compute_and_compare_results
+
+class TestQlassf(unittest.TestCase):
+    def test_print_call(self):
+        f = "def test(a: bool) -> bool:\n\tprint(a)\n\treturn a"
+        qf = qlassf(f, to_compile=COMPILATION_ENABLED)
+        self.assertEqual(len(qf.expressions), 1)
+        compute_and_compare_results(self, qf)
 
 class TestQlassfDecorator(unittest.TestCase):
     def test_decorator(self):

tox.ini

@@ -31,7 +31,7 @@ deps =
     pytest
     parameterized
 commands =
-    pytest -rP
+    pytest #-rP
 
 [testenv:flake8]
 deps =