pytorch

1
## @package onnx
2
# Module caffe2.python.onnx.tests.helper_test
3

4

5

6

7

8

9
import unittest
10

11
from caffe2.python.onnx.tests.test_utils import TestCase
12
import caffe2.python._import_c_extension as C
13

14

15
class TestCaffe2Basic(TestCase):
16
    def test_dummy_name(self):
17
        g = C.DummyName()
18
        g.reset()
19
        names_1 = [g.new_dummy_name() for _ in range(3)]
20
        g.reset()
21
        names_2 = [g.new_dummy_name() for _ in range(3)]
22
        self.assertEqual(names_1, names_2)
23

24
        g.reset(set(names_1))
25
        names_3 = [g.new_dummy_name() for _ in range(3)]
26
        self.assertFalse(set(names_1) & set(names_3))
27

28
        g.reset(set(names_1))
29
        names_4 = [g.new_dummy_name() for _ in range(3)]
30
        self.assertFalse(set(names_1) & set(names_4))
31

32

33
if __name__ == '__main__':
34
    unittest.main()
35