wip

Author: czgdp1807

integration_tests/CMakeLists.txt

@@ -152,7 +152,8 @@ RUN(NAME test_list_02        LABELS cpython llvm)
 RUN(NAME test_list_03        LABELS cpython llvm)
 RUN(NAME test_list_04        LABELS cpython llvm)
 RUN(NAME test_list_05        LABELS cpython llvm)
-RUN(NAME test_list_06        LABELS llvm)
+RUN(NAME test_list_06        LABELS cpython llvm)
+RUN(NAME test_list_07        LABELS cpython llvm)
 RUN(NAME test_tuple_01       LABELS cpython llvm)
 RUN(NAME test_tuple_02       LABELS cpython llvm)
 RUN(NAME modules_01          LABELS cpython llvm)

integration_tests/test_list_06.py

@@ -1,4 +1,5 @@
 from ltypes import i32, f64
+from copy import deepcopy
 
 def check_mat_and_vec(mat: list[list[f64]], vec: list[f64]):
     rows: i32 = len(mat)
@@ -28,7 +29,7 @@ def test_list_of_lists():
     for i in range(rows):
         for j in range(cols):
             vec.append(float(i + j))
-        mat.append(vec)
+        mat.append(deepcopy(vec))
         vec.clear()
 
     for i in range(cols):
@@ -37,7 +38,7 @@ def test_list_of_lists():
     check_mat_and_vec(mat, vec)
 
     for k in range(rows):
-        tensor.append(mat)
+        tensor.append(deepcopy(mat))
         for i in range(rows):
             for j in range(cols):
                 mat[i][j] += float(1)
@@ -48,7 +49,7 @@ def test_list_of_lists():
                 assert mat[i][j] - tensor[k][i][j] == rows - k
 
     for l in range(2 * rows):
-        tensors.append(tensor)
+        tensors.append(deepcopy(tensor))
         for i in range(rows):
             for j in range(rows):
                 for k in range(cols):

integration_tests/test_list_07.py

@@ -1,10 +1,11 @@
 from ltypes import c64, i32
+from copy import deepcopy
 
 def test_tuple_with_lists():
     mat: list[list[c64]] = []
     vec: list[c64] = []
     tensor: tuple[list[list[c64]], list[c64]]
-    tensors: list[tuple[list[list[c64]], list[c64]]]
+    tensors: list[tuple[list[list[c64]], list[c64]]] = []
     i: i32
     j: i32
     k: i32
@@ -15,7 +16,7 @@ def test_tuple_with_lists():
     for i in range(rows):
         for j in range(cols):
             vec.append(complex(i + j, 0))
-        mat.append(vec)
+        mat.append(deepcopy(vec))
         vec.clear()
 
     for i in range(cols):
@@ -25,7 +26,7 @@ def test_tuple_with_lists():
         for j in range(cols):
             assert mat[i][j] - vec[j] == i - j
 
-    tensor = (mat, vec)
+    tensor = (deepcopy(mat), deepcopy(vec))
 
     for i in range(rows):
         for j in range(cols):
@@ -41,18 +42,18 @@ def test_tuple_with_lists():
     for i in range(cols):
         assert tensor[1][i] - vec[i] == -complex(0, 2.0)
 
-    tensor = (mat, vec)
+    tensor = (deepcopy(mat), deepcopy(vec))
 
     for k in range(2 * rows):
-        tensors.append(tensor)
+        tensors.append(deepcopy(tensor))
         for i in range(rows):
             for j in range(cols):
                 mat[i][j] += complex(1.0, 2.0)
 
         for i in range(cols):
             vec[i] += complex(1.0, 2.0)
 
-        tensor = (mat, vec)
+        tensor = (deepcopy(mat), deepcopy(vec))
 
     for k in range(2 * rows):
         for i in range(rows):

src/lpython/semantics/python_ast_to_asr.cpp

@@ -197,6 +197,9 @@ ASR::Module_t* load_module(Allocator &al, SymbolTable *symtab,
                             std::vector<std::string> &rl_path,
                             bool &ltypes,
                             const std::function<void (const std::string &, const Location &)> err) {
+    if( module_name == "copy" ) {
+        return nullptr;
+    }
     ltypes = false;
     LFORTRAN_ASSERT(symtab);
     if (symtab->get_scope().find(module_name) != symtab->get_scope().end()) {
@@ -4422,6 +4425,14 @@ class BodyVisitor : public CommonVisitor<BodyVisitor> {
                                         ASRUtils::type_to_str(type) + " type.", x.base.base.loc);
                 }
                 return;
+            } else if( call_name == "deepcopy" ) {
+                if( args.size() != 1 ) {
+                    throw SemanticError("deepcopy only accepts one argument, found " +
+                                        std::to_string(args.size()) + " instead.",
+                                        x.base.base.loc);
+                }
+                tmp = (ASR::asr_t*) args[0].m_value;
+                return ;
             } else {
                 // The function was not found and it is not intrinsic
                 throw SemanticError("Function '" + call_name + "' is not declared and not intrinsic",

src/lpython/semantics/python_attribute_eval.h

@@ -189,9 +189,17 @@ struct AttributeHandler {
     }
 
     static ASR::asr_t* eval_list_clear(ASR::expr_t *s, Allocator &al,
-        const Location &loc, Vec<ASR::expr_t*> &args, diag::Diagnostics &diag) {
+        const Location &loc, Vec<ASR::expr_t*> &args, diag::Diagnostics & diag) {
             if (args.size() != 0) {
-                throw SemanticError("clear() takes no argument", loc);
+                diag.add(diag::Diagnostic(
+                    "Incorrect number of arguments in 'clear', it accepts no argument",
+                    diag::Level::Error, diag::Stage::Semantic, {
+                        diag::Label("incorrect number of arguments in clear (found: " +
+                                    std::to_string(args.size()) + ", expected: 0)",
+                                {loc})
+                    })
+                );
+                throw SemanticAbort();
             }
 
             return make_ListClear_t(al, loc, s);