1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
# Owner(s): ["module: onnx"]
"""Unit tests for the _tensors module."""
from __future__ import annotations
import onnxscript
from torch.onnx._internal.exporter import _tensors
from torch.testing._internal import common_utils
class SymbolicTensorTest(common_utils.TestCase):
def test_it_is_hashable(self):
tensor = _tensors.SymbolicTensor(
opset=onnxscript.values.Opset(domain="test", version=1)
)
self.assertEqual(hash(tensor), hash(tensor))
self.assertIn(tensor, {tensor})
if __name__ == "__main__":
common_utils.run_tests()
|
