pytorch

1
import unittest
2

3
from caffe2.python import workspace
4

5

6
# This test is extracted out from workspace_test.py because it relies on the pristine
7
# state of the initial workspace. When tests are run in different orders, this test may
8
# become flaky because of global state modifications impacting what the root folder is
9
# after a reset.
10
class TestWorkspace(unittest.TestCase):
11
    def testRootFolder(self):
12
        self.assertEqual(workspace.ResetWorkspace(), True)
13
        self.assertEqual(workspace.RootFolder(), ".")
14
        self.assertEqual(workspace.ResetWorkspace("/tmp/caffe-workspace-test"), True)
15
        self.assertEqual(workspace.RootFolder(), "/tmp/caffe-workspace-test")
16