Merge pull request #124 from namannimmo10/set

Completes initial type annotations

Author: namannimmo10

src/lpython/semantics/python_ast_to_asr.cpp

@@ -216,6 +216,28 @@ class CommonVisitor : public AST::BaseVisitor<Derived> {
                 ASR::ttype_t *type = ASRUtils::TYPE(ASR::make_Tuple_t(al, loc,
                     types.p, types.size()));
                 return type;
+            } else if (var_annotation == "set") {
+                if (AST::is_a<AST::Name_t>(*s->m_slice)) {
+                    ASR::ttype_t *type = ast_expr_to_asr_type(loc, *s->m_slice);
+                    return ASRUtils::TYPE(ASR::make_Set_t(al, loc, type));
+                } else {
+                    throw SemanticError("Only Name in Subscript supported for now in `set`"
+                        " annotation", loc);
+                }
+            } else if (var_annotation == "dict") {
+                if (AST::is_a<AST::Tuple_t>(*s->m_slice)) {
+                    AST::Tuple_t *t = AST::down_cast<AST::Tuple_t>(s->m_slice);
+                    if (t->n_elts != 2) {
+                        throw SemanticError("`dict` annotation must have 2 elements: types"
+                            " of both keys and values", loc);
+                    }
+                    ASR::ttype_t *key_type = ast_expr_to_asr_type(loc, *t->m_elts[0]);
+                    ASR::ttype_t *value_type = ast_expr_to_asr_type(loc, *t->m_elts[1]);
+                    return ASRUtils::TYPE(ASR::make_Dict_t(al, loc, key_type, value_type));
+                } else {
+                    throw SemanticError("`dict` annotation must have 2 elements: types of"
+                        " both keys and values", loc);
+                }
             } else {
                 ASR::dimension_t dim;
                 dim.loc = loc;

src/runtime/ltypes/ltypes.py

@@ -2,3 +2,4 @@
 i64 = []
 f32 = []
 f64 = []
+c128 = []