import wandb

from wandb import data_types

import numpy as np

import pytest

import PIL

import os

import six

import sys

import glob

import platform

import pandas as pd

from click.testing import CliRunner

from . import utils

import matplotlib

matplotlib.use("Agg")

import matplotlib.pyplot as plt # noqa: E402

data = np.random.randint(255, size=(1000))

def test_raw_data():

wbhist = wandb.Histogram(data)

assert len(wbhist.histogram) == 64

def test_np_histogram():

wbhist = wandb.Histogram(np_histogram=np.histogram(data))

assert len(wbhist.histogram) == 10

def test_manual_histogram():

wbhist = wandb.Histogram(np_histogram=([1, 2, 4], [3, 10, 20, 0]))

assert len(wbhist.histogram) == 3

def test_invalid_histogram():

with pytest.raises(ValueError):

wandb.Histogram(np_histogram=([1, 2, 3], [1]))

image = np.zeros((28, 28))

def test_captions():

wbone = wandb.Image(image, caption="Cool")

wbtwo = wandb.Image(image, caption="Nice")

assert wandb.Image.all_captions([wbone, wbtwo]) == ["Cool", "Nice"]

def test_bind_image(mocked_run):

wb_image = wandb.Image(image)

wb_image.bind_to_run(mocked_run, "stuff", 10)

assert wb_image.is_bound()

full_box = {

}

def dissoc(d, key):

new_d = d.copy()

new_d.pop(key)

return new_d

optional_keys = ["box_caption", "scores"]

boxes_with_removed_optional_args = [dissoc(full_box, k) for k in optional_keys]

def test_image_accepts_other_images(mocked_run):

image_a = wandb.Image(np.random.random((300, 300, 3)))

image_b = wandb.Image(image_a)

assert image_a == image_b

def test_image_accepts_bounding_boxes(mocked_run):

assert os.path.exists(os.path.join(mocked_run.dir, path))

def test_image_accepts_bounding_boxes_optional_args(mocked_run):

assert os.path.exists(os.path.join(mocked_run.dir, path))

standard_mask = {

"mask_data": np.array([[1, 2, 2, 2], [2, 3, 3, 4], [4, 4, 4, 4], [4, 4, 4, 2]]),

"class_labels": {1: "car", 2: "pedestrian", 3: "tractor", 4: "cthululu"},

}

def test_image_accepts_masks(mocked_run):

assert os.path.exists(os.path.join(mocked_run.dir, path))

def test_image_accepts_masks_without_class_labels(mocked_run):

assert os.path.exists(os.path.join(mocked_run.dir, path))

def test_cant_serialize_to_other_run(mocked_run, test_settings):

wb_image.to_json(other_run)

def test_image_seq_to_json(mocked_run):

assert utils.subdict(meta, meta_expected) == meta_expected

def test_max_images(caplog, mocked_run):

assert os.path.exists(os.path.join(mocked_run.dir, "media/images/test2_0_0.png"))

def test_audio_sample_rates():

wandb.Audio(audio1)

def test_audio_durations():

assert wandb.Audio.durations([wbaudio1, wbaudio2]) == [1.0, 2.0]

def test_audio_captions():

assert wandb.Audio.captions([wbaudio5, wbaudio6]) == ["", caption2]

def test_audio_to_json(mocked_run):

assert utils.subdict(meta["audio"][0], audio_expected) == audio_expected

def test_guess_mode():

image = np.random.randint(255, size=(28, 28, 3))

wbimg = wandb.Image(image)

assert wbimg._image.mode == "RGB"

def test_pil():

pil = PIL.Image.new("L", (28, 28))

img = wandb.Image(pil)

assert img._image == pil

def test_matplotlib_image():

plt.plot([1, 2, 2, 4])

img = wandb.Image(plt)

assert img._image.width == 640

def test_matplotlib_image_with_multiple_axes():

wandb.Image(plt) # this should not error.

def test_matplotlib_plotly_with_multiple_axes():

wandb.Plotly(plt) # this should not error.

def test_plotly_from_matplotlib_with_image():

plt.close()

def test_image_from_matplotlib_with_image():

plt.close()

def test_make_plot_media_from_matplotlib_without_image():

plt.close()

def test_make_plot_media_from_matplotlib_with_image():

plt.close()

def test_video_numpy(mocked_run):

assert vid.to_json(mocked_run)["path"].endswith(".gif")

def test_video_numpy_multi(mocked_run):

assert vid.to_json(mocked_run)["path"].endswith(".gif")

def test_video_numpy_invalid():

video = np.random.random(size=(3, 28, 28))

with pytest.raises(ValueError):

wandb.Video(video)

def test_video_path(mocked_run):

assert vid.to_json(mocked_run)["path"].endswith(".mp4")

def test_video_path_invalid(runner):

wandb.Video("video.avi")

def test_molecule(runner, mocked_run):

assert os.path.exists(mol._path)

def test_html_str(mocked_run):

assert os.path.exists(html._path)

def test_html_styles():

assert html.html == "<h1>Hello</h1>"

def test_html_file(mocked_run):

assert os.path.exists(html._path)

def test_table_default():

}

def test_table_custom():

assert table._to_table_json() == table_df._to_table_json()

point_cloud_1 = np.array([[0, 0, 0, 1], [0, 0, 1, 13], [0, 1, 0, 2], [0, 1, 0, 4]])

point_cloud_2 = np.array([[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 0]])

point_cloud_3 = np.array(

)

def test_object3d_numpy(mocked_run):

assert obj3.to_json(mocked_run)["_type"] == "object3D-file"

def test_object3d_obj(mocked_run):

obj = wandb.Object3D(utils.fixture_open("cube.obj"))

obj.bind_to_run(mocked_run, "object3D", 0)

assert obj.to_json(mocked_run)["_type"] == "object3D-file"

def test_object3d_gltf(mocked_run):

obj = wandb.Object3D(utils.fixture_open("Box.gltf"))

obj.bind_to_run(mocked_run, "object3D", 0)

assert obj.to_json(mocked_run)["_type"] == "object3D-file"

def test_object3d_io(mocked_run):

assert obj.to_json(mocked_run)["_type"] == "object3D-file"

def test_object3d_unsupported_numpy():

wandb.Object3D(ioObj)

def test_object3d_seq_to_json(mocked_run):

]

def test_table_init():

}

def test_graph():

}

def test_numpy_arrays_to_list():

assert conv(np.array(({"a": [np.array((1, 2,))]}, 3))) == [{"a": [[1, 2]]}, 3]