def test_heatmap_inputs(): """ test parameter checks """ X = np.random.rand(10, 10) with pytest.raises(TypeError): heatmap(X="input") # transform with pytest.raises(ValueError): transform = "bad transform" heatmap(X, transform=transform) # cmap with pytest.raises(TypeError): cmap = 123 heatmap(X, cmap=cmap) # center with pytest.raises(TypeError): center = "center" heatmap(X, center=center) # cbar with pytest.raises(TypeError): cbar = 1 heatmap(X, cbar=cbar)
def test_heatmap_output(): """ simple function to see if plot is made without errors """ X = er_np(10, 0.5) xticklabels = ["Dimension {}".format(i) for i in range(10)] yticklabels = ["Dimension {}".format(i) for i in range(10)] fig = heatmap(X, transform="log", xticklabels=xticklabels, yticklabels=yticklabels) fig = heatmap(X, transform="zero-boost") fig = heatmap(X, transform="simple-all") fig = heatmap(X, transform="simple-nonzero") fig = heatmap(X, cmap="gist_rainbow")
def test_common_inputs(): X = er_np(100, 0.5) grid_labels = ["Test1"] # test figsize with pytest.raises(TypeError): figsize = "bad figsize" heatmap(X, figsize=figsize) # test height height = "1" with pytest.raises(TypeError): gridplot([X], grid_labels, height=height) with pytest.raises(TypeError): pairplot(X, height=height) # test title title = 1 with pytest.raises(TypeError): heatmap(X, title=title) with pytest.raises(TypeError): gridplot([X], grid_labels, title=title) with pytest.raises(TypeError): pairplot(X, title=title) # test context context = 123 with pytest.raises(TypeError): heatmap(X, context=context) with pytest.raises(TypeError): gridplot([X], grid_labels, context=context) with pytest.raises(TypeError): pairplot(X, context=context) context = "journal" with pytest.raises(ValueError): heatmap(X, context=context) with pytest.raises(ValueError): gridplot([X], grid_labels, context=context) with pytest.raises(ValueError): pairplot(X, context=context) # test font scales font_scales = ["1", []] for font_scale in font_scales: with pytest.raises(TypeError): heatmap(X, font_scale=font_scale) with pytest.raises(TypeError): gridplot([X], grid_labels, font_scale=font_scale) with pytest.raises(TypeError): pairplot(X, cont_scale=font_scale) # ticklabels with pytest.raises(TypeError): xticklabels = "labels" yticklabels = "labels" heatmap(X, xticklabels=xticklabels, yticklabels=yticklabels) with pytest.raises(ValueError): xticklabels = ["{}".format(i) for i in range(5)] yticklabels = ["{}".format(i) for i in range(5)] heatmap(X, xticklabels=xticklabels, yticklabels=yticklabels)