Added top-level unloading of currently selected API version, MPI version, and path

Author: Martin Ruefenacht

src/pympistandard/__init__.py

@@ -17,7 +17,7 @@
 import os
 
 
-from .storage import KINDS, PROCEDURES, CALLBACKS, PREDEFINED_FUNCTIONS
+from .storage import KINDS, PROCEDURES, CALLBACKS, PREDEFINED_FUNCTIONS, clear_storage
 from .parameter import Direction
 from .procedure import Procedure
 from .callback import Callback
@@ -45,13 +45,19 @@
 )
 
 
+@export
+def unload() -> None:
+    """Unloads currently loaded Python API Interface and MPI Standard."""
+
+    clear_storage()
+
+
+# TODO rename to load(api_version, mpi_version, path)
 @export
 def use_api_version(version: Union[int, str] = "LATEST", given_path: Union[str, Path] = None) -> None:
     """Sets the Python API interface which the user expects to use."""
 
-    # clean from possible prior usage
-    KINDS.clear()
-    PROCEDURES.clear()
+    unload()
 
     # load version of API
     if version in (1, "LATEST"):

src/pympistandard/storage.py

@@ -20,3 +20,10 @@
 CONSTANTS = NamespaceDict()
 # to support this the Python DSL in the MPI Standard needs to be extended
 # and all the constants need to be encoded (name, type, value if given)
+
+def clear_storage():
+    KINDS.clear()
+    PROCEDURES.clear()
+    CALLBACKS.clear()
+    PREDEFINED_FUNCTIONS.clear()
+    CONSTANTS.clear()