(tests) Improved tearDown of customModel to improve model-related memory handling in the testing pipeline

Author: amkrajewski

pysipfenn/tests/test_ModelAdjusters.py

@@ -50,6 +50,7 @@ def tearDown(self):
                 delattr(self, attr)
         # Core objects created in setUp — order matters: adjuster holds a ref into the Calculator
         del self.ma
+        self.c.destroy()
         del self.c
         gc.collect()
 

pysipfenn/tests/test_ModelExporters.py

@@ -34,6 +34,7 @@ def tearDown(self):
             if hasattr(self, attr):
                 delattr(self, attr)
         # Core Calculator created in setUp
+        self.c.destroy()
         del self.c
         gc.collect()
 

pysipfenn/tests/test_customModel.py

@@ -1,4 +1,5 @@
 import unittest
+import gc
 from pymatgen.core import Structure
 from importlib.resources import files as resources_files, as_file
 import shutil
@@ -47,8 +48,10 @@ def testCalculation(self):
 
     @pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Test depends on the ONNX network files")
     def tearDown(self) -> None:
-        '''Deletes the copied model.'''
-        self.c = None
+        '''Deletes the copied model and frees the Calculator.'''
+        self.c.destroy()
+        del self.c
+        gc.collect()
         print('\nTearing down')
         os.remove('MyFunNet.onnx')
         print('Removed MyFunNet')