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,)}, ], }
wb_image = wandb.Image(image) try: wb_image.bind_to_run(mocked_run, "my key: an identifier", 0) if platform.system() == "Windows": assert False except ValueError as e: assert " is invalid. Please remove invalid filename characters" in str(e) runbindable_media = [ wandb.Image(image, masks={"overlay": standard_mask}), wandb.data_types.ImageMask( {"mask_data": np.random.randint(0, 10, (300, 300))}, key="test" ), wandb.Table(data=[[1, 2, 3], [4, 5, 6]]), wandb.Graph(), wandb.Audio(np.random.uniform(-1, 1, 44100), sample_rate=44100), ] @pytest.mark.parametrize("media", runbindable_media) def test_media_keys_escaped_as_glob_for_publish(mocked_run, media): weird_key = "[weirdkey]" media.bind_to_run(mocked_run, weird_key, 0) published_globs = [ g for ( [files_dict], [], ) in mocked_run._backend.interface.publish_files.call_args_list for g, _ in files_dict["files"]