def test_o_multiple_outputs(self):
x = Variable(name="x")
y = Variable(name="y")
y2 = Variable(name="y2")
node = Node(op="Add", name="Input", inputs=[x], outputs=[y])
node2 = Node(op="Add", name="Input", inputs=[x], outputs=[y2])
assert x.o() == y
assert x.o(1) == y2
def test_o(self):
x = Variable(name="x")
y = Variable(name="y")
node = Node(op="Add", name="Input", inputs=[x], outputs=[y])
assert x.o() == y
