def test_serialize_node_serialize_to_pickle(clear_default_graph):
node1 = SquareNode('Node1')
node2 = SquareNode(name='Node2')
node1.inputs['in1'].value = 1
node1.outputs['out'] >> node2.inputs['in1']
node1.outputs['compound_out']['key'] >> node2.inputs['compound_in']['key']
node1.outputs['compound_out']['1'] >> node2.inputs['compound_in']['1']
node1.outputs['out'].value = 'value'
node1.outputs['compound_out']['key'].value = 'value_key'
node1.outputs['compound_out']['1'].value = 'value_1'
node2.outputs['out'].value = 'value'
node2.outputs['compound_out']['key'].value = 'value_key'
node2.outputs['compound_out']['1'].value = 'value_1'
rec1 = INode.from_pickle(node1.to_pickle())
# Using the to_json serialization as a check here is suboptimal, but we
# don't have another INode.__eq__(self, other)
assert rec1.to_json() == node1.to_json()
rec2 = INode.from_pickle(node2.to_pickle())
assert rec2.to_json() == node2.to_json()
def test_serialize_node_serialize_to_pickle(clear_default_graph):
node1 = SquareNode("Node1")
node2 = SquareNode(name="Node2")
node1.inputs["in1"].value = 1
node1.outputs["out"] >> node2.inputs["in1"]
node1.outputs["compound_out"]["key"] >> node2.inputs["compound_in"]["key"]
node1.outputs["compound_out"]["1"] >> node2.inputs["compound_in"]["1"]
node1.outputs["out"].value = "value"
node1.outputs["compound_out"]["key"].value = "value_key"
node1.outputs["compound_out"]["1"].value = "value_1"
node2.outputs["out"].value = "value"
node2.outputs["compound_out"]["key"].value = "value_key"
node2.outputs["compound_out"]["1"].value = "value_1"
rec1 = INode.from_pickle(node1.to_pickle())
# Using the to_json serialization as a check here is suboptimal, but we
# don't have another INode.__eq__(self, other)
assert rec1.to_json() == node1.to_json()
rec2 = INode.from_pickle(node2.to_pickle())
assert rec2.to_json() == node2.to_json()