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(): 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)
def test_pairplot_inputs(): X = np.random.rand(15, 3) Y = ["A"] * 5 + ["B"] * 5 + ["C"] * 5 # test data with pytest.raises(TypeError): pairplot(X="test") with pytest.raises(ValueError): pairplot(X=X, labels=["A"]) with pytest.raises(TypeError): pairplot(X, col_names="A") with pytest.raises(ValueError): pairplot(X, col_names=["1", "2"]) with pytest.raises(ValueError): pairplot(X, col_names=["1", "2", "3"], variables=[1, 2, 3, 4]) with pytest.raises(KeyError): pairplot(X, col_names=["1", "2", "3"], variables=["A", "B"])