From a22daa2d06c502fe52f8acc79b592102e3390a44 Mon Sep 17 00:00:00 2001
From: nucccc <dan.dom.nuc@gmail.com>
Date: Wed, 27 Mar 2024 21:01:58 +0100
Subject: [PATCH] getting type from tuple

---
 .../convert/collect/ast_to_typ/ast_to_typ.py  | 17 +++++++++
 tests/test_ast_to_typ.py                      | 36 +++++++++++++++++--
 2 files changed, 51 insertions(+), 2 deletions(-)

diff --git a/markarth/convert/collect/ast_to_typ/ast_to_typ.py b/markarth/convert/collect/ast_to_typ/ast_to_typ.py
index ba0c0e5..f349b11 100644
--- a/markarth/convert/collect/ast_to_typ/ast_to_typ.py
+++ b/markarth/convert/collect/ast_to_typ/ast_to_typ.py
@@ -22,6 +22,8 @@ def ast_val_to_typ(
         return typ_from_constant(val)
     if type(val) == ast.BinOp:
         return typ_from_bin_op(val, var_tracker)
+    if type(val) == ast.Tuple:
+        return typ_from_tuple(ast_tup = val, var_tracker = var_tracker)
     if var_tracker is not None:
         if type(val) == ast.Name:
             supposed_typ = var_tracker.get_vartyp( val.id )
@@ -123,6 +125,7 @@ def typ_from_call(
             return call_type
     return typs.TypAny()
 
+
 def typ_from_iter(iter_stat : ast.AST) -> typs.Typ:
     '''
     type_from_iter shall get the type of a variable "extracted"
@@ -139,3 +142,17 @@ def typ_from_iter(iter_stat : ast.AST) -> typs.Typ:
                 typs.TypAny()
             ])
     return typs.TypAny()
+
+
+def typ_from_tuple(
+    ast_tup : ast.Tuple,
+    var_tracker : VarTypTracker | None = None
+) -> typs.TypTuple:
+    '''
+    typ_from_tuple returns a tuple typ from an ast tuple statement
+    '''
+    # for each of the tuple's elements i gather their typ
+    inner_typs = [
+        ast_val_to_typ(elt, var_tracker) for elt in ast_tup.elts
+    ]
+    return typs.TypTuple(inner_typs = inner_typs)
diff --git a/tests/test_ast_to_typ.py b/tests/test_ast_to_typ.py
index 96bb718..b6eb7f6 100644
--- a/tests/test_ast_to_typ.py
+++ b/tests/test_ast_to_typ.py
@@ -7,8 +7,10 @@
     typ_from_iter,
     typ_from_call,
     ast_val_to_typ,
-    eval_op_typ
+    eval_op_typ,
+    typ_from_tuple
 )
+from markarth.convert.collect.vartyp_tracker import VarTypTracker
 from markarth.convert.typs import typs
 
 
@@ -226,4 +228,34 @@ def test_typ_from_call():
     call = mod.body[0].value
 
     call_typ = typ_from_call(call = call)
-    assert call_typ.is_any()
\ No newline at end of file
+    assert call_typ.is_any()
+
+
+def test_typ_from_tuple():
+    code = 'a = (4, 5, 6.11)'
+    mod = ast.parse(code)
+    ast_tup = mod.body[0].value
+
+    typ = typ_from_tuple(ast_tup = ast_tup)
+
+    assert typ.is_tuple()
+    assert len(typ.inner_typs) == 3
+    assert typ.inner_typs[0].is_int()
+    assert typ.inner_typs[1].is_int()
+    assert typ.inner_typs[2].is_float()
+
+
+def test_typ_from_tuple_with_vartracker():
+    code = 'a = (4, a, 6.11)'
+    mod = ast.parse(code)
+    ast_tup = mod.body[0].value
+    vartyp_tracker = VarTypTracker()
+    vartyp_tracker.add_local_typ('a', typs.TypPrimitive(typs.PrimitiveCod.BOOL))
+
+    typ = typ_from_tuple(ast_tup = ast_tup, var_tracker = vartyp_tracker)
+
+    assert typ.is_tuple()
+    assert len(typ.inner_typs) == 3
+    assert typ.inner_typs[0].is_int()
+    assert typ.inner_typs[1].is_bool()
+    assert typ.inner_typs[2].is_float()
\ No newline at end of file