def test_pairplot_outputs():
X = np.random.rand(15, 3)
Y = ["A"] * 5 + ["B"] * 5 + ["C"] * 5
col_names = ["Feature1", "Feature2", "Feature3"]
fig = pairplot(X)
fig = pairplot(X, Y)
fig = pairplot(X, Y, col_names)
fig = pairplot(
X, Y, col_names, title="Test", height=1.5, variables=["Feature1", "Feature2"]
)
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_pairplot_inputs(self):
X = np.random.rand(15, 3)
Y = ["A"] * 5 + ["B"] * 5 + ["C"] * 5
# test data
with self.assertRaises(TypeError):
pairplot(X="test")
with self.assertRaises(ValueError):
pairplot(X=X, labels=["A"])
with self.assertRaises(TypeError):
pairplot(X, col_names="A")
with self.assertRaises(ValueError):
pairplot(X, col_names=["1", "2"])
with self.assertRaises(ValueError):
pairplot(X, col_names=["1", "2", "3"], variables=[1, 2, 3, 4])
with self.assertRaises(KeyError):
pairplot(X, col_names=["1", "2", "3"], variables=["A", "B"])
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")
