def test_heatmap_inputs(self): """ test parameter checks """ X = np.random.rand(10, 10) with self.assertRaises(TypeError): heatmap(X="input") # transform with self.assertRaises(ValueError): transform = "bad transform" heatmap(X, transform=transform) # cmap with self.assertRaises(TypeError): cmap = 123 heatmap(X, cmap=cmap) # center with self.assertRaises(TypeError): center = "center" heatmap(X, center=center) # cbar with self.assertRaises(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, transform="binarize") fig = heatmap(X, cmap="gist_rainbow")
def test_common_inputs(self): X = er_np(100, 0.5) x = np.random.rand(100, 1) y = np.random.rand(100, 1) grid_labels = ["Test1"] # test figsize figsize = "bad figsize" with self.assertRaises(TypeError): heatmap(X, figsize=figsize) with self.assertRaises(beartype.roar.BeartypeCallHintParamViolation): with self.assertRaises(TypeError): networkplot(adjacency=X, x=x, y=y, figsize=figsize) # test height height = "1" with self.assertRaises(TypeError): gridplot([X], grid_labels, height=height) with self.assertRaises(TypeError): pairplot(X, height=height) # test title title = 1 with self.assertRaises(TypeError): heatmap(X, title=title) with self.assertRaises(TypeError): gridplot([X], grid_labels, title=title) with self.assertRaises(TypeError): pairplot(X, title=title) with self.assertRaises(beartype.roar.BeartypeCallHintParamViolation): with self.assertRaises(TypeError): networkplot(adjacency=X, x=x, y=y, title=title) # test context context = 123 with self.assertRaises(TypeError): heatmap(X, context=context) with self.assertRaises(TypeError): gridplot([X], grid_labels, context=context) with self.assertRaises(TypeError): pairplot(X, context=context) with self.assertRaises(beartype.roar.BeartypeCallHintParamViolation): with self.assertRaises(TypeError): networkplot(adjacency=X, x=x, y=y, context=context) context = "journal" with self.assertRaises(ValueError): heatmap(X, context=context) with self.assertRaises(ValueError): gridplot([X], grid_labels, context=context) with self.assertRaises(ValueError): pairplot(X, context=context) with self.assertRaises(ValueError): networkplot(adjacency=X, x=x, y=y, context=context) # test font scales font_scales = ["1", []] for font_scale in font_scales: with self.assertRaises(TypeError): heatmap(X, font_scale=font_scale) with self.assertRaises(TypeError): gridplot([X], grid_labels, font_scale=font_scale) with self.assertRaises(TypeError): pairplot(X, font_scale=font_scale) with self.assertRaises( beartype.roar.BeartypeCallHintParamViolation): with self.assertRaises(TypeError): networkplot(adjacency=X, x=x, y=y, font_scale=font_scale) # ticklabels with self.assertRaises(TypeError): xticklabels = "labels" yticklabels = "labels" heatmap(X, xticklabels=xticklabels, yticklabels=yticklabels) with self.assertRaises(ValueError): xticklabels = ["{}".format(i) for i in range(5)] yticklabels = ["{}".format(i) for i in range(5)] heatmap(X, xticklabels=xticklabels, yticklabels=yticklabels) with self.assertRaises(TypeError): heatmap(X, title_pad="f") with self.assertRaises(TypeError): gridplot([X], title_pad="f") with self.assertRaises(TypeError): heatmap(X, hier_label_fontsize="f") with self.assertRaises(TypeError): gridplot([X], hier_label_fontsize="f")
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) with pytest.raises(TypeError): heatmap(X, title_pad="f") with pytest.raises(TypeError): gridplot([X], title_pad="f") with pytest.raises(TypeError): heatmap(X, hier_label_fontsize="f") with pytest.raises(TypeError): gridplot([X], hier_label_fontsize="f")