Iterate over string

Vipul-Cariappa · Jun 24, 2023 · a374001 · a374001
1 parent 4662acb
commit a374001
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/integration_tests/test_list_11.py b/integration_tests/test_list_11.py
@@ -11,12 +11,20 @@ def test_issue_1882():
     for i in [1, 2, 3]:
         assert i + 1 == x[i - 1]
 
+def test_iterate_over_string():
+    s: str
+    i: i32 = 0
+    temp: str = "abcd"
+    for s in "abcd":
+        assert s == temp[i]
+        i+=1
 
 def main0():
     x: list[i32] = return_empty_list_of_tuples()
     print(len(x))
 
     assert len(x) == 0
     test_issue_1882()
+    test_iterate_over_string()
 
 main0()
diff --git a/src/lpython/semantics/python_ast_to_asr.cpp b/src/lpython/semantics/python_ast_to_asr.cpp
@@ -5371,7 +5371,7 @@ class BodyVisitor : public CommonVisitor<BodyVisitor> {
                 throw SemanticError("Only Name is supported for Subscript",
                     sbt->base.base.loc);
             }
-        } else if (AST::is_a<AST::List_t>(*x.m_iter)) {
+        } else if (AST::is_a<AST::List_t>(*x.m_iter) || AST::is_a<AST::ConstantStr_t>(*x.m_iter) ) {
             visit_expr(*x.m_iter);
             ASR::expr_t *target = ASRUtils::EXPR(tmp);
             ASR::ttype_t *loop_src_var_ttype = ASRUtils::expr_type(target);