mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
1022443168
Test Plan: revert-hammer
Differential Revision:
D30279364 (b004307252)
Original commit changeset: c1ed77dfe43a
fbshipit-source-id: eab50857675c51e0088391af06ec0ecb14e2347e
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
|
|
|
|
|
|
|
|
|
|
import errno
|
|
import os
|
|
import tempfile
|
|
import shutil
|
|
|
|
from caffe2.distributed.python import StoreHandlerTimeoutError
|
|
from caffe2.distributed.store_ops_test_util import StoreOpsTests
|
|
from caffe2.python import core, workspace, dyndep
|
|
from caffe2.python.test_util import TestCase
|
|
|
|
dyndep.InitOpsLibrary("@/caffe2/caffe2/distributed:file_store_handler_ops")
|
|
dyndep.InitOpsLibrary("@/caffe2/caffe2/distributed:store_ops")
|
|
|
|
|
|
class TestFileStoreHandlerOp(TestCase):
|
|
testCounter = 0
|
|
|
|
def setUp(self):
|
|
super(TestFileStoreHandlerOp, self).setUp()
|
|
self.tmpdir = tempfile.mkdtemp()
|
|
|
|
# Use counter to tell test cases apart
|
|
TestFileStoreHandlerOp.testCounter += 1
|
|
|
|
def tearDown(self):
|
|
shutil.rmtree(self.tmpdir)
|
|
super(TestFileStoreHandlerOp, self).tearDown()
|
|
|
|
def create_store_handler(self):
|
|
# Use new path for every test so they are isolated
|
|
path = self.tmpdir + "/" + str(TestFileStoreHandlerOp.testCounter)
|
|
|
|
# Ensure path exists (including counter)
|
|
try:
|
|
os.makedirs(path)
|
|
except OSError as exc:
|
|
if exc.errno == errno.EEXIST and os.path.isdir(path):
|
|
pass
|
|
else:
|
|
raise
|
|
|
|
store_handler = "store_handler"
|
|
workspace.RunOperatorOnce(
|
|
core.CreateOperator(
|
|
"FileStoreHandlerCreate",
|
|
[],
|
|
[store_handler],
|
|
path=path))
|
|
|
|
return store_handler
|
|
|
|
def test_set_get(self):
|
|
StoreOpsTests.test_set_get(self.create_store_handler)
|
|
|
|
def test_get_timeout(self):
|
|
with self.assertRaises(StoreHandlerTimeoutError):
|
|
StoreOpsTests.test_get_timeout(self.create_store_handler)
|