def test_outputs(self):
op = Operator()
t0 = Tensor()
t0.index = 0
t1 = Tensor()
t1.index = 1
op.outputs = [t0, t1]
self.assertEqual(len(op.outputs), 2)
self.assertEqual(op.outputs[0], t0)
self.assertEqual(op.outputs[1], t1)
def test_tensors_map(self):
subg = Subgraph()
t0 = Tensor()
t0.index = 0
t1 = Tensor()
t1.index = 1
subg.tensors_map[t0.index] = t0
subg.tensors_map[t1.index] = t1
self.assertEqual(len(subg.tensors_map.keys()), 2)
self.assertEqual(subg.tensors_map[t0.index], t0)
self.assertEqual(subg.tensors_map[t1.index], t1)
def test_outputs(self):
subg = Subgraph()
t0 = Tensor()
t0.index = 0
t1 = Tensor()
t1.index = 1
subg.outputs = [t0, t1]
self.assertEqual(len(subg.outputs), 2)
self.assertEqual(subg.outputs[0], t0)
self.assertEqual(subg.outputs[0].index, 0)
self.assertEqual(subg.outputs[1], t1)
self.assertEqual(subg.outputs[1].index, 1)
def test_memory_size(self):
t = Tensor()
t.memory_size = 1000
self.assertEqual(t.memory_size, 1000)
def test_shape(self):
t = Tensor()
t.shape = [1, 2, 3, 4]
self.assertEqual(t.shape, [1, 2, 3, 4])
def test_type_name(self):
t = Tensor()
t.type_name = "FLOAT32"
self.assertEqual(t.type_name, "FLOAT32")
def test_buffer_index(self):
t = Tensor()
t.buffer_index = 1000
self.assertEqual(t.buffer_index, 1000)
def test_buffer(self):
t = Tensor()
o = object()
t.buffer = o
self.assertEqual(t.buffer, o)
def test_tensor_name(self):
t = Tensor()
t.tensor_name = "input"
self.assertEqual(t.tensor_name, "input")