Added tests

advikkabra · advikkabra · commit 657df67f36e2 · 2024-06-26T11:49:08.000+05:30
diff --git a/integration_tests/CMakeLists.txt b/integration_tests/CMakeLists.txt
@@ -584,10 +584,12 @@ RUN(NAME test_dict_bool      LABELS cpython llvm llvm_jit)
 RUN(NAME test_dict_increment LABELS cpython llvm llvm_jit)
 RUN(NAME test_dict_keys_values LABELS cpython llvm llvm_jit)
 RUN(NAME test_dict_nested1   LABELS cpython llvm llvm_jit)
+RUN(NAME test_dict_clear     LABELS cpython llvm)
 RUN(NAME test_set_len        LABELS cpython llvm llvm_jit)
 RUN(NAME test_set_add        LABELS cpython llvm llvm_jit)
 RUN(NAME test_set_remove     LABELS cpython llvm llvm_jit)
 RUN(NAME test_set_discard     LABELS cpython llvm llvm_jit)
+RUN(NAME test_set_clear     LABELS cpython llvm)
 RUN(NAME test_global_set     LABELS cpython llvm llvm_jit)
 RUN(NAME test_for_loop       LABELS cpython llvm llvm_jit c)
 RUN(NAME modules_01          LABELS cpython llvm llvm_jit c wasm wasm_x86 wasm_x64)
diff --git a/integration_tests/test_dict_clear.py b/integration_tests/test_dict_clear.py
@@ -0,0 +1,20 @@
+def test_clear():
+    a: dict[i32, i32] = {1:1, 2:2}
+
+    a.clear()
+    a[3] = 3
+
+    assert len(a) == 1
+    assert a.keys() == [3]
+    assert a.values() == [3]
+
+    b: dict[str, str] = {'a':'a', 'b':'b'}
+
+    b.clear()
+    b['c'] = 'c'
+
+    assert len(b) == 1
+    assert b.keys() == ['c']
+    assert b.values() == ['c']
+
+test_clear()
diff --git a/integration_tests/test_set_clear.py b/integration_tests/test_set_clear.py
@@ -0,0 +1,21 @@
+def test_clear():
+    a: set[i32] = {1, 2}
+
+    a.clear()
+    a.add(3)
+
+    assert len(a) == 1
+    # a.remove(3)
+    # assert len(a) == 0
+
+    b: set[str] = {'a', 'b'}
+
+    b.clear()
+    b.add('c')
+
+    assert len(b) == 1
+    # b.remove('c')
+    # assert len(b) == 0
+
+
+test_clear()
diff --git a/src/libasr/codegen/llvm_utils.cpp b/src/libasr/codegen/llvm_utils.cpp
@@ -6926,16 +6926,16 @@ namespace LCompilers {
         llvm::Value* new_el_mask = LLVM::lfortran_calloc(context, *module, *builder, llvm_zero,
                                                           llvm_mask_size);
         std::string el_type_code = ASRUtils::get_type_code(el_asr_type);
-        //llvm::Type* el_llvm_type = std::get<2>(typecode2settype[el_type_code]);
-        //int32_t el_type_size = std::get<1>(typecode2settype[el_type_code]);
+        llvm::Type* el_llvm_type = std::get<2>(typecode2settype[el_type_code]);
+        int32_t el_type_size = std::get<1>(typecode2settype[el_type_code]);
 
-        //llvm::Value* new_el_list = builder0.CreateAlloca(llvm_utils->list_api->get_list_type(el_llvm_type,
-                                                          //el_type_code, el_type_size), nullptr);
+        llvm::Value* new_el_list = builder0.CreateAlloca(llvm_utils->list_api->get_list_type(el_llvm_type,
+                                                          el_type_code, el_type_size), nullptr);
         llvm_utils->list_api->list_init(el_type_code, el_list, *module, llvm_zero, llvm_zero);
 
         llvm_utils->list_api->free_data(el_list, *module);
         LLVM::lfortran_free(context, *module, *builder, LLVM::CreateLoad(*builder, get_pointer_to_mask(set)));
-        //LLVM::CreateStore(*builder, LLVM::CreateLoad(*builder, new_el_list), el_list);
+        LLVM::CreateStore(*builder, LLVM::CreateLoad(*builder, new_el_list), el_list);
         LLVM::CreateStore(*builder, new_el_mask, get_pointer_to_mask(set));
     }
 
