Make tests thread-safe.

python/lsst/ts/wep/CreatePhosimDonutTemplates.py

@@ -51,6 +51,19 @@ def __init__(self):
         # The location where we will store the final templates
         self.templateDestDir = os.path.join(self.templateDataPath, "phosimTemplates")
 
+    def setTempWorkPaths(self, newBasePath):
+        """
+        Point to a new temporary work directory. Update work repo as well.
+
+        Parameters
+        ----------
+        newBasePath : str
+            New location of temporary work directories.
+        """
+
+        self.tempWorkPath = newBasePath
+        self.repoDir = os.path.join(self.tempWorkPath, 'input')
+
     def createWorkDirectories(self):
         """
         Create the final template directory as well as

tests/test_createPhosimDonutTemplates.py

@@ -22,6 +22,7 @@
 import unittest
 import os
 import shutil
+import tempfile
 import numpy as np
 
 from lsst.ts.wep.CreatePhosimDonutTemplates import CreatePhosimDonutTemplates
@@ -41,9 +42,9 @@ def setUp(self):
         # Location where templates are created
         self.templateDir = os.path.join(self.templateDataDir, "phosimTemplates")
         # Temporary work directory
-        self.tempWorkDir = os.path.join(self.templateDataDir, "tempDir")
-        # Make the temporary directory for all tests
-        os.mkdir(self.tempWorkDir)
+        self.tempTestDirectory = tempfile.TemporaryDirectory()
+        self.tempWorkDir = self.tempTestDirectory.name
+        self.createPhosimDonuts.setTempWorkPaths(self.tempWorkDir)
 
     def tearDown(self):
 
@@ -54,7 +55,7 @@ def tearDown(self):
             pass
 
         # Remove work directory in case of clean up method test failure
-        shutil.rmtree(self.tempWorkDir)
+        self.tempTestDirectory.cleanup()
 
     def _copyPhosimFiles(self):