Set support : copy method (pyccel#1765)

Add support for the `copy()` method of Python sets to the semantic stage.
Add handling of that method to the Python printer. This fixes  pyccel#1740.

Author: mustapha-belbiad

CHANGELOG.md

@@ -7,6 +7,7 @@ All notable changes to this project will be documented in this file.
 
 -   #1739 : Add Python support for set method `clear()`.
 -   #1739 : Add abstract class `SetMethod` to handle calls to various set methods.
+-   #1740 : Add Python support for set method `copy()`.
 
 ### Fixed
 

pyccel/ast/builtin_methods/set_methods.py

@@ -12,7 +12,8 @@
 from pyccel.ast.datatypes import NativeVoid, NativeGeneric
 from pyccel.ast.internals import PyccelInternalFunction
 
-__all__ = ('SetAdd', 'SetClear', 'SetMethod')
+__all__ = ('SetAdd', 'SetClear', 'SetMethod', 'SetCopy')
+
 
 class SetMethod(PyccelInternalFunction):
     """
@@ -26,7 +27,7 @@ class SetMethod(PyccelInternalFunction):
     set_variable : TypedAstNode
         The set on which the method will operate.
 
-    *args : iterable
+    *args : TypedAstNode
         The arguments passed to the function call.
     """
     __slots__ = ('_set_variable',)
@@ -44,6 +45,7 @@ def set_variable(self):
         """
         return self._set_variable
 
+
 class SetAdd(SetMethod) :
     """
     Represents a call to the .add() method.
@@ -81,6 +83,7 @@ def __init__(self, set_variable, new_elem) -> None:
             raise TypeError("Expecting an argument of the same type as the elements of the set")
         super().__init__(set_variable, new_elem)
 
+
 class SetClear(SetMethod):
     """
     Represents a call to the .clear() method.
@@ -104,3 +107,28 @@ class SetClear(SetMethod):
 
     def __init__(self, set_variable):
         super().__init__(set_variable)
+
+
+class SetCopy(SetMethod):
+    """
+    Represents a call to the .copy() method.
+
+    The copy() method in set class creates a shallow 
+    copy of a set object and returns it. 
+
+    Parameters
+    ----------
+    set_variable : TypedAstNode
+        The set on which the method will operate.
+    """
+    __slots__ = ("_dtype","_shape", "_order", "_rank", "_precision", "_class_type",)
+    name = 'copy'
+
+    def __init__(self, set_variable):
+        self._dtype = set_variable._dtype
+        self._shape = set_variable._shape
+        self._order = set_variable._order
+        self._rank = set_variable._rank
+        self._precision = set_variable._precision
+        self._class_type = set_variable._class_type
+        super().__init__(set_variable)

pyccel/ast/class_defs.py

@@ -6,7 +6,7 @@
 This module contains all types which define a python class which is automatically recognised by pyccel
 """
 
-from pyccel.ast.builtin_methods.set_methods import SetAdd, SetClear
+from pyccel.ast.builtin_methods.set_methods import SetAdd, SetClear, SetCopy
 from pyccel.ast.builtin_methods.list_methods import ListAppend, ListInsert, ListPop, ListClear
 
 
@@ -153,10 +153,9 @@
 
 SetClass = ClassDef('set', class_type=NativeHomogeneousSet(),
         methods=[
-            PyccelFunctionDef('add', func_class = SetAdd,
-                decorators = {}),
-            PyccelFunctionDef('clear', func_class = SetClear,
-                decorators={}),
+            PyccelFunctionDef('add', func_class = SetAdd ),
+            PyccelFunctionDef('clear', func_class = SetClear),
+            PyccelFunctionDef('copy', func_class = SetCopy),
         ])
 
 #=======================================================================================

tests/epyccel/test_epyccel_sets.py

@@ -77,3 +77,40 @@ def clear_complex():
     python_result = clear_complex()
     assert python_result == pyccel_result
 
+def test_copy_int(language):
+    def copy_int():
+        se = {1, 2, 4, 5}
+        cop = se.copy()
+        return cop
+    epyccel_copy = epyccel(copy_int, language = language)
+    pyccel_result = epyccel_copy()
+    python_result = copy_int()
+    assert isinstance(python_result, type(pyccel_result))
+    assert python_result == pyccel_result
+    assert all(isinstance(elem, type(pyccel_result.pop())) for elem in python_result)
+
+
+def test_copy_float(language):
+    def copy_float():
+        se = {5.7, 6.2, 4.3, 9.8}
+        cop = se.copy()
+        return cop
+    epyccel_copy = epyccel(copy_float, language = language)
+    pyccel_result = epyccel_copy()
+    python_result = copy_float()
+    assert isinstance(python_result, type(pyccel_result))
+    assert python_result == pyccel_result
+    assert all(isinstance(elem, type(pyccel_result.pop())) for elem in python_result)
+
+def test_copy_complex(language):
+    def copy_complex():
+        se = {7j, 6j, 9j}
+        cop = se.copy()
+        return cop
+    epyccel_copy = epyccel(copy_complex, language = language)
+    pyccel_result = epyccel_copy()
+    python_result = copy_complex()
+    assert isinstance(python_result, type(pyccel_result))
+    assert python_result == pyccel_result
+    assert all(isinstance(elem, type(pyccel_result.pop())) for elem in python_result)
+