def test_i_multiple_inputs(self):
intermediate_tensor = Variable(name="intermediate")
intermediate_tensor2 = Variable(name="intermediate2")
input_node = Node(op="Add", name="Input", inputs=[self.input_tensor], outputs=[intermediate_tensor])
input_node2 = Node(op="Add", name="Input2", inputs=[self.input_tensor], outputs=[intermediate_tensor2])
output_node = Node(op="Add", name="Out", inputs=[intermediate_tensor, intermediate_tensor2], outputs=[self.output_tensor])
assert output_node.i() == input_node
assert output_node.i(1) == input_node2
def test_i(self):
intermediate_tensor = Variable(name="intermediate")
input_node = Node(op="Add",
name="Input",
inputs=[self.input_tensor],
outputs=[intermediate_tensor])
output_node = Node(op="Add",
name="Out",
inputs=[intermediate_tensor],
outputs=[self.output_tensor])
assert output_node.i() == input_node
