Test a few more features

mmoskal · mmoskal · commit 7f2ed86a50e5 · 2019-03-20T18:11:07.000-07:00
diff --git a/pxtpy/converter.ts b/pxtpy/converter.ts
@@ -14,6 +14,8 @@ namespace pxt.py {
     let ctx: Ctx
     let currIteration = 0
     let typeId = 0
+    // this measures if we gained additional information about type state
+    // we run conversion several times, until we have all information possible
     let numUnifies = 0
     let currErrs = ""
     let autoImport = true
@@ -1414,6 +1416,7 @@ namespace pxt.py {
             if (!formals) {
                 if (fun)
                     error(n, U.lf("calling non-function"))
+                allargs = orderedArgs.map(expr)
             } else {
                 if (orderedArgs.length > formals.length)
                     error(n, U.lf("too many arguments in call to '{0}'", fun.qName))
diff --git a/pxtpy/parser.ts b/pxtpy/parser.ts
@@ -248,6 +248,7 @@ namespace pxt.py {
         let r = mkAST("For") as For
         expectKw("for")
         r.target = exprlist()
+        setStoreCtx(r.target)
         expectKw("in")
         r.iter = testlist()
         r.body = colon_suite()
@@ -958,6 +959,7 @@ namespace pxt.py {
             rr.push(r)
             expectKw("for")
             r.target = exprlist()
+            setStoreCtx(r.target)
             expectKw("in")
             r.iter = or_test()
             r.ifs = []
diff --git a/tests/python/game.py b/tests/python/game.py
@@ -3,6 +3,12 @@ def qux2(self):
     z = 12
     x = z * 3
     self.baz = x
+    for q in range(10):
+      x += q
+    lst = ["foo", "bar", "baz"]
+    lst = lst[1:2]
+    # TODO change to len(lst)
+    assert lst.length == 2, 201
   def qux(self):
     self.baz = self.bar
     self.blah = "hello"
