| | |
| | |
| |
|
| |
|
| |
|
| |
|
| |
|
| |
|
| | import unittest |
| |
|
| | from caffe2.python.onnx.tests.test_utils import TestCase |
| | import caffe2.python._import_c_extension as C |
| |
|
| |
|
| | class TestCaffe2Basic(TestCase): |
| | def test_dummy_name(self): |
| | g = C.DummyName() |
| | g.reset() |
| | names_1 = [g.new_dummy_name() for _ in range(3)] |
| | g.reset() |
| | names_2 = [g.new_dummy_name() for _ in range(3)] |
| | self.assertEqual(names_1, names_2) |
| |
|
| | g.reset(set(names_1)) |
| | names_3 = [g.new_dummy_name() for _ in range(3)] |
| | self.assertFalse(set(names_1) & set(names_3)) |
| |
|
| | g.reset(set(names_1)) |
| | names_4 = [g.new_dummy_name() for _ in range(3)] |
| | self.assertFalse(set(names_1) & set(names_4)) |
| |
|
| |
|
| | if __name__ == '__main__': |
| | unittest.main() |
| |
|
