Close ELMo resources in tests

plasticityai · Nov 8, 2018 · f56383f · f56383f
1 parent 6ae8d57
commit f56383f
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/pymagnitude/__init__.py b/pymagnitude/__init__.py
@@ -1956,6 +1956,20 @@ def close(self):
                 self.APPROX_MMAP_PROCESS_LOCK.lockfile.close()
             except BaseException:
                 pass
+        if (hasattr(self, 'ELMO_W_MMAP_PROCESS_LOCK') and
+            hasattr(self.ELMO_W_MMAP_PROCESS_LOCK, 'lockfile') and
+                self.ELMO_W_MMAP_PROCESS_LOCK.lockfile is not None):
+            try:
+                self.ELMO_W_MMAP_PROCESS_LOCK.lockfile.close()
+            except BaseException:
+                pass
+        if (hasattr(self, 'ELMO_O_MMAP_PROCESS_LOCK') and
+            hasattr(self.ELMO_O_MMAP_PROCESS_LOCK, 'lockfile') and
+                self.ELMO_O_MMAP_PROCESS_LOCK.lockfile is not None):
+            try:
+                self.ELMO_O_MMAP_PROCESS_LOCK.lockfile.close()
+            except BaseException:
+                pass
 
     def __del__(self):
         """ Destructor for the class """

diff --git a/tests/tests.py b/tests/tests.py
@@ -91,10 +91,20 @@ def setUp(self):
 
     def tearDown(self):
         self.vectors.close()
+        self.vectors_un.close()
         self.vectors_cs.close()
         self.vectors_batch.close()
         self.vectors_sw.close()
+        self.vectors_approx.close()
         self.tmp_vectors.close()
+        try:
+            self.vectors_elmo.close()
+            self.vectors_elmo_p.close()
+            self.vectors_elmo_ngram.close()
+            self.vectors_elmo_ngram_p.close()
+            self.vectors_elmo_n.close()
+        except BaseException:
+            pass
         self.concat_1.close()
         self.concat_2.close()
         del self.concat