def test_i_multiple_inputs(self):
x = Variable(name="x")
x2 = Variable(name="x2")
y = Variable(name="y")
node = Node(op="Add", name="Input", inputs=[x, x2], outputs=[y])
assert y.i() == x
assert y.i(1) == x2
def test_i(self):
x = Variable(name="x")
y = Variable(name="y")
node = Node(op="Add", name="Input", inputs=[x], outputs=[y])
assert y.i() == x