Store annotated local variables to avoid memory leaks (pyccel#1904)

Add annotated local variable to the `_allocs` variable so its
deallocation node can be added in `_garbage_collector()`. This fixes
pyccel#1903

Author: Farouk-Echaref

CHANGELOG.md

@@ -36,6 +36,7 @@ All notable changes to this project will be documented in this file.
 -   #1853 : Fix translation of a file whose name conflicts with Fortran keywords.
 -   Link and mention `devel` branch, not `master`.
 -   #1047 : Print the value of an unrecognised constant.
+-   #1903 : Fix memory leak when using type annotations on local variables.
 
 ### Changed
 

pyccel/parser/semantic.py

@@ -1683,7 +1683,9 @@ def _ensure_inferred_type_matches_existing(self, class_type, d_var, var, is_auga
 
                     new_expressions.append(Allocate(var, shape=d_var['shape'], status=status))
 
-                    if status != 'unallocated':
+                    if status == 'unallocated':
+                        self._allocs[-1].add(var)
+                    else:
                         errors.report(ARRAY_REALLOCATION, symbol=var.name,
                             severity='warning',
                             bounding_box=(self.current_ast_node.lineno,

tests/ndarrays/leaks_check.py

@@ -1,5 +1,5 @@
 # pylint: disable=missing-function-docstring, missing-module-docstring
-from numpy import array, zeros
+from numpy import array, zeros, ones
 
 def create_array():
     a = array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])# pylint: disable=unused-variable
@@ -45,6 +45,9 @@ def arrays_in_multi_returns():
     b = zeros(4)
     return a, b, 4
 
+def type_annotated_numpy_array():
+    a : 'float[:]' = ones(5) # pylint: disable=unused-variable
+
 # testing garbage collecting in a Function
 
 if __name__ == '__main__':
@@ -56,6 +59,7 @@ def arrays_in_multi_returns():
     conditional_alloc(True,True)
     conditional_alloc(True,False)
     conditional_alloc(False,False)
+    type_annotated_numpy_array()
 
     # testing garbage collecting in a Program