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