def test_single_tensor(self):
X = {0: ['v']}
X_x_X = {0: {('v', 'v')}}
self.maxDiff = None
self.assertEqual(tensor(X, X), X_x_X)
def test_multidim_tensor(self):
X = {0: ['v'],
1: ['a', 'b'],
2: ['aa', 'ab']}
X_x_X = {0: {('v', 'v')},
1: {('v', 'a'), ('v', 'b'), ('a', 'v'), ('b', 'v')},
2: {('v', 'aa'), ('v', 'ab'), ('aa', 'v'), ('ab', 'v'),
('a', 'b'), ('b', 'a'), ('a', 'a'), ('b', 'b')},
3: {('a', 'aa'), ('a', 'ab'), ('b', 'aa'), ('b', 'ab'),
('aa', 'a'), ('aa', 'b'), ('ab', 'a'), ('ab', 'b')},
4: {('aa', 'aa'), ('aa', 'ab'), ('ab', 'aa'), ('ab', 'ab')}}
self.maxDiff = None
self.assertEqual(tensor(X, X), X_x_X)
