added Type hints (#7) for the Instruction Classes

scripts/Infrastructure/AllocCall.py

@@ -24,7 +24,7 @@ alloc_template = """
 
 class AllocCall(Instruction):
     @override
-    def __init__(self, type, num_elements, name="buf", use_malloc=False):
+    def __init__(self, type: str, num_elements: int, name: str = "buf", use_malloc: bool = False):
         """
         Creates a New allocation Call
 
@@ -55,13 +55,13 @@ class AllocCall (Instruction):
                 .replace("@{NUM}@", str(self._num_elements))
                 .replace("@{SEP}@", delim))
 
-    def set_num_elements(self, num_elements):
+    def set_num_elements(self, num_elements: int):
         self._num_elements = num_elements
 
-    def set_name(self, name):
+    def set_name(self, name: str):
         self._name = name
 
-    def set_type(self, type):
+    def set_type(self, type: str):
         self._type = type
 
     def set_use_malloc(self):
@@ -70,19 +70,19 @@ class AllocCall (Instruction):
     def set_use_calloc(self):
         self._use_malloc = False
 
-    def get_num_elements(self):
+    def get_num_elements(self) -> int:
         return self._num_elements
 
-    def get_name(self):
+    def get_name(self) -> str:
         return self._name
 
-    def get_type(self):
+    def get_type(self) -> str:
         return self._type
 
-    def get_use_malloc(self):
+    def get_use_malloc(self) -> bool:
         return self._use_malloc
 
 
-def get_free(alloc_call):
+def get_free(alloc_call: AllocCall) -> Instruction:
     assert isinstance(alloc_call, AllocCall)
-    return "free(" + alloc_call.get_name() + ");\n"
+    return Instruction("free(" + alloc_call.get_name() + ");")

scripts/Infrastructure/Instruction.py

@@ -10,4 +10,4 @@ class Instruction(object):
         self._str_representation = str_representation
 
     def __str__(self):
-        return self._str_representation
+        return self._str_representation + "\n"

scripts/Infrastructure/MPICall.py

 #! /usr/bin/python3
+import typing
 from typing_extensions import override
 
 from scripts.Infrastructure.Instruction import Instruction
@@ -8,7 +9,7 @@ from scripts.Infrastructure.Variables import ERROR_MARKER_COMMENT
 class MPI_Call(Instruction):
 
     @override
-    def __init__(self, function, args, version):
+    def __init__(self, function: str, args: typing.OrderedDict[str, str], version: str):
         self._function = function
         self._args = args
         self._version = version
@@ -27,16 +28,15 @@ class MPI_Call (Instruction):
 
         return s
 
-    def set_arg(self, arg, value):
+    def set_arg(self, arg: str, value: str):
         assert self.has_arg(arg)
         self._args[arg] = value
 
-    def has_arg(self, arg):
+    def has_arg(self, arg: str) -> bool:
         return arg in self._args
 
-    def get_version(self):
+    def get_version(self) -> str:
         return self._version
 
-    def set_has_error(self, has_error=True):
+    def set_has_error(self, has_error: bool = True):
         self._has_error = has_error
-