diff --git a/caffe2/python/clean_workspace_test.py b/caffe2/python/clean_workspace_test.py new file mode 100644 index 00000000000..c8285f4a1c5 --- /dev/null +++ b/caffe2/python/clean_workspace_test.py @@ -0,0 +1,15 @@ +import unittest + +from caffe2.python import workspace + + +# This test is extracted out from workspace_test.py because it relies on the pristine +# state of the initial workspace. When tests are run in different orders, this test may +# become flaky because of global state modifications impacting what the root folder is +# after a reset. +class TestWorkspace(unittest.TestCase): + def testRootFolder(self): + self.assertEqual(workspace.ResetWorkspace(), True) + self.assertEqual(workspace.RootFolder(), ".") + self.assertEqual(workspace.ResetWorkspace("/tmp/caffe-workspace-test"), True) + self.assertEqual(workspace.RootFolder(), "/tmp/caffe-workspace-test") diff --git a/caffe2/python/workspace_test.py b/caffe2/python/workspace_test.py index 2e2d284f92e..b434b5e748c 100644 --- a/caffe2/python/workspace_test.py +++ b/caffe2/python/workspace_test.py @@ -24,12 +24,6 @@ class TestWorkspace(unittest.TestCase): ) workspace.ResetWorkspace() - def testRootFolder(self): - self.assertEqual(workspace.ResetWorkspace(), True) - self.assertEqual(workspace.RootFolder(), ".") - self.assertEqual(workspace.ResetWorkspace("/tmp/caffe-workspace-test"), True) - self.assertEqual(workspace.RootFolder(), "/tmp/caffe-workspace-test") - def testWorkspaceHasBlobWithNonexistingName(self): self.assertEqual(workspace.HasBlob("non-existing"), False)