def test_graph(): graph = wandb.Graph() node_a = data_types.Node('a', 'Node A', size=(4,)) node_b = data_types.Node('b', 'Node B', size=(16,)) graph.add_node(node_a) graph.add_node(node_b) graph.add_edge(node_a, node_b) assert graph._to_graph_json() == { 'edges': [['a', 'b']], 'format': 'keras', 'nodes': [{'id': 'a', 'name': 'Node A', 'size': (4,)}, {'id': 'b', 'name': 'Node B', 'size': (16,)}]}
def test_graph(): graph = wandb.Graph() node_a = data_types.Node("a", "Node A", size=(4,)) node_b = data_types.Node("b", "Node B", size=(16,)) graph.add_node(node_a) graph.add_node(node_b) graph.add_edge(node_a, node_b) assert graph._to_graph_json() == { "edges": [["a", "b"]], "format": "keras", "nodes": [ {"id": "a", "name": "Node A", "size": (4,)}, {"id": "b", "name": "Node B", "size": (16,)}, ], }