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,)},
],
}