Issue #983 | Add Round function and implement the conversion to C, Fortran, and Python

pyccel/ast/builtins.py

@@ -42,6 +42,7 @@
     'PythonZip',
     'PythonMax',
     'PythonMin',
+    'PythonRound',
     'python_builtin_datatype'
 )
 
@@ -319,6 +320,26 @@ def arg(self):
     def __str__(self):
         return 'float({0})'.format(str(self.arg))
 
+# ===========================================================================
+class PythonRound(PyccelAstNode):
+    """ Represents a call to Python's native round() function.
+    """
+    name = 'round'
+    _dtype = NativeReal()
+    _precision = default_precision['real']
+    _rank = 0
+    _shape = ()
+    _order = None
+    _attribute_nodes  = ('_arg',)
+
+    def __init__(self, arg):
+        self._arg = arg
+        super().__init__()
+
+    @property
+    def arg(self):
+        return self._arg
+
 #==============================================================================
 class PythonInt(PyccelAstNode):
     """ Represents a call to Python's native int() function.
@@ -861,4 +882,5 @@ def python_builtin_datatype(name):
     'not'      : PyccelNot,
     'map'      : PythonMap,
     'type'     : PythonType,
+    'round'    : PythonRound,
 }

pyccel/codegen/printing/ccode.py

@@ -419,6 +419,11 @@ def _print_PythonAbs(self, expr):
             func = "labs"
         return "{}({})".format(func, self._print(expr.arg))
 
+    def _print_PythonRound(self, expr):
+        self._additional_imports.add("math")
+        func = "round"
+        return "{}({})".format(func, self._print(expr.arg))
+
     def _print_PythonMin(self, expr):
         arg = expr.args[0]
         if arg.dtype is NativeReal() and len(arg) == 2:

pyccel/codegen/printing/fcode.py

@@ -543,6 +543,12 @@ def _print_PythonAbs(self, expr):
         """
         return "abs({})".format(self._print(expr.arg))
 
+    def _print_PythonRound(self, expr):
+        """ print the python builtin function round
+        args : variable
+        """
+        return "nint({})".format(self._print(expr.arg))
+
     def _print_PythonTuple(self, expr):
         shape = tuple(reversed(expr.shape))
         if len(shape)>1:

pyccel/codegen/printing/pycode.py

@@ -372,6 +372,9 @@ def _print_PythonImag(self, expr):
     def _print_PythonPrint(self, expr):
         return 'print({})\n'.format(', '.join(self._print(a) for a in expr.expr))
 
+    def _print_PythonRound(self, expr):
+        return 'round({})'.format(self._print(expr.arg))
+
     def _print_PyccelArraySize(self, expr):
         arg = self._print(expr.arg)
         index = self._print(expr.index)

tests/epyccel/test_epyccel_round.py

@@ -0,0 +1,17 @@
+# pylint: disable=missing-function-docstring, missing-module-docstring/
+from numpy.random import randint
+
+from pyccel.decorators import types
+from pyccel.epyccel import epyccel
+
+def test_round_int(language):
+    @types('real')
+    def round_int(x):
+        return round(x)
+
+    f = epyccel(round_int, language=language)
+    x = randint(100) / 10
+
+    f_output = f(x)
+    round_int_output = round_int(x)
+    assert round_int_output == f_output