| from __future__ import absolute_import |
| from __future__ import division |
| from __future__ import print_function |
| from __future__ import unicode_literals |
| from caffe2.python import context, test_util |
| from threading import Thread |
| @context.define_context() |
| class TestContext(test_util.TestCase): |
| def use_my_context(self): |
| self.assertTrue(MyContext.current() == a) |
| self._exceptions.append(e) |
| def testMultiThreaded(self): |
| thread = Thread(target=self.use_my_context) |
| for e in self._exceptions: |
| self.assertIsNotNone(MyContext.current()) |