Restore working directory after test

salilab · Feb 27, 2024 · efe83d4 · efe83d4
1 parent a5b182a
commit efe83d4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/modules/spatiotemporal/test/test_make_graph.py b/modules/spatiotemporal/test/test_make_graph.py
@@ -44,6 +44,7 @@ def test_graph_setup(self):
             # keys correspond to all timepoints
             keys = list(state_dict.keys())
             # Go to input_dir, if it exists
+            orig_dir = os.getcwd()
             if len(input_dir) > 0:
                 if os.path.exists(input_dir):
                     os.chdir(input_dir)
@@ -73,6 +74,8 @@ def test_graph_setup(self):
             # check that all nodes are graphNode objects
             for node in nodes:
                 self.assertIsInstance(node,graphNode.graphNode)
+            # Restore original working directory
+            os.chdir(orig_dir)
 
     def test_graph_scoring(self):
         """