def test_dbscan_labels_colored(self): '''Points with different color should be two clusters with rgb_weight = 1''' labels = dbscan_labels(self.pc, 0.1, 1, rgb_weight=1) labelcount = len(np.unique(labels)) assert_equals(labelcount, 2)
def test_dbscan_labels(self): '''Points with different color should be one cluster with rgb_weight = 0''' labels = dbscan_labels(self.pc, 0.1, 1, rgb_weight=0) labelcount = len(np.unique(labels)) assert_equals(labelcount, 1)