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"]