def test_serialize_graph_to_json(clear_default_graph): """ +------------+ +------------+ +--------------------+ | Start | | Node2 | | End | |------------| |------------| |--------------------| o in1<0> | +--->o in1<> | % in1 | o in2<0> | | o in2<0> | +--->o in1.1<> | | out o-----+ | out o-----|--->o in1.2<> | | out2 o | | out2 o | o in2<0> | +------------+ | +------------+ | | out o | +------------+ | | out2 o | | Node1 | | +--------------------+ | |------------| | +--->o in1<> | | o in2<0> | | | out o-----+ | out2 o +------------+ """ graph = Graph() start = NodeForTesting(name='Start', graph=graph) n1 = NodeForTesting(name='Node1', graph=graph) n2 = NodeForTesting(name='Node2', graph=graph) end = NodeForTesting(name='End', graph=graph) start.outputs['out'] >> n1.inputs['in1'] start.outputs['out'] >> n2.inputs['in1'] n1.outputs['out'] >> end.inputs['in1']['1'] n2.outputs['out'] >> end.inputs['in1']['2'] serialized = graph.to_json() deserialized = Graph.from_json(serialized).to_json() assert serialized == deserialized
def test_serialize_graph_to_json(clear_default_graph, branching_graph): serialized = branching_graph.to_json() deserialized = Graph.from_json(serialized) assert branching_graph.name == deserialized.name assert serialized == deserialized.to_json()