def test_cluster_scatter_grad_lab_diff_custom_markers(self): sorted_x = self.x_50x2[np.argsort(self.x_50x2[:, 1])] # should not have error even if gradient is provided eda.cluster_scatter(sorted_x, labels=[0] * 25 + [1] * 25, shuffle_label_colors=True, gradient=sorted_x[:, 1], title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) fig = eda.cluster_scatter(sorted_x, labels=[0] * 25 + [1] * 25, gradient=sorted_x[:, 1], plot_different_markers=True, label_markers=['o'] * 10 + ['*'] * 25 + ['^'] * 15, title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) return fig
def test_cluster_scatter_no_xylab_title(self): fig = eda.cluster_scatter(self.x_50x2, [0] * 25 + [1] * 10 + [2] * 15, figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) return fig
def test_cluster_scatter_shuffle_labcol(self): fig = eda.cluster_scatter(self.x_50x2, [0] * 25 + [1] * 10 + [2] * 15, shuffle_label_colors=True, figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=2) return fig
def test_cluster_scatter_no_randstate(self): eda.cluster_scatter(self.x_50x2, [0] * 25 + [1] * 25, title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50) eda.cluster_scatter(self.x_50x2, [0] * 25 + [1] * 25, title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, random_state=None, s=50)
def test_cluster_scatter_diff_lab_markers(self): fig = eda.cluster_scatter(self.x_50x2, [0] * 25 + [1] * 10 + [2] * 15, figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, plot_different_markers=True, s=50, random_state=123) return fig
def test_cluster_scatter_no_labels(self): fig = eda.cluster_scatter(self.x_50x2, title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) return fig
def test_cluster_scatter_no_legends(self): fig = eda.cluster_scatter(self.x_50x2, [0] * 25 + [1] * 25, title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), add_legend=False, n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) return fig
def test_cluster_scatter_gradient_legends_nolab(self): sorted_x = self.x_50x2[np.argsort(self.x_50x2[:, 1])] fig = eda.cluster_scatter(sorted_x, labels=None, gradient=sorted_x[:, 1], title='test tsne scatter', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) return fig
def test_cluster_scatter_xylim(self): fig = eda.cluster_scatter(np.array( [list(range(50)), list(range(100, 150))]).T, [0] * 25 + [1] * 25, title='xlim=(25, 50), ylim=(120, 150)', figsize=(10, 10), add_legend=False, xlim=(25, 50), ylim=(120, 150), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) return fig
def test_cluster_scatter_nogradient_legends_slab0(self): sorted_x = self.x_50x2[np.argsort(self.x_50x2[:, 1])] fig = eda.cluster_scatter(sorted_x, labels=[0] * 25 + [1] * 25, selected_labels=[0], title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) return fig
def test_cluster_scatter_diff_lab_markers_wrong_args(self): # labs not provided with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(self.x_50x2, figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, plot_different_markers=True, label_markers=['o'] * 10 + ['*'] * 25 + ['^'] * 15, s=50, random_state=123) # labs and markers have different lengths # wrong marker length with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(self.x_50x2, labels=[0] * 25 + [1] * 10 + [2] * 15, figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, plot_different_markers=True, label_markers=['o'] * 10 + ['*'] * 25 + ['^'] * 10, s=50, random_state=123) # wrong label length with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(self.x_50x2, labels=[0] * 25 + [1] * 10 + [2] * 10, figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, plot_different_markers=True, label_markers=['o'] * 10 + ['*'] * 25 + ['^'] * 15, s=50, random_state=123)
def test_cluster_scatter_wrong_args(self): # wrong projection2d dimention with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(np.random.ranf(100).reshape(-1, 1), title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(np.random.ranf(100).reshape(-1, 5), title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(np.random.ranf(99).reshape(-1, 3), title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) # wrong label shape with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(self.x_50x2, [0] * 60, title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(self.x_50x2, [[0]] * 50, title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) # wrong gradient shape with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(self.x_50x2, [0] * 50, gradient=list(range(60)), title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(self.x_50x2, [[0]] * 50, gradient=np.arange(50).reshape(-1, 1), title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) # select labels without providing labels with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(self.x_50x2, selected_labels=[0], gradient=list(range(50)), title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) # select absent labels with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(self.x_50x2, [0] * 50, selected_labels=[1], gradient=list(range(50)), title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123) with pytest.raises(ValueError) as excinfo: eda.cluster_scatter(self.x_50x2, [0] * 50, selected_labels=[1, 0], gradient=list(range(50)), title='test tsne scatter', xlab='tsne1', ylab='tsne2', figsize=(10, 10), n_txt_per_cluster=3, alpha=0.5, s=50, random_state=123)