def test_clustering_DBSCAN_direct(self, ens1): pytest.importorskip('sklearn') method = encore.DBSCAN(eps=0.5, min_samples=2) distance_matrix = encore.get_distance_matrix(ens1) cluster_assignment, details = method(distance_matrix) expected_value = 2 assert len(set(cluster_assignment)) == expected_value, \ "Unexpected result: {0}".format(cluster_assignment)
def test_clustering_DBSCAN_direct(self): method = encore.DBSCAN(eps=0.5, min_samples=2) distance_matrix = encore.get_distance_matrix(self.ens1) cluster_assignment, details = method(distance_matrix) expected_value = 2 assert_equal( len(set(cluster_assignment)), expected_value, err_msg="Unexpected result: {0}".format(cluster_assignment))
def test_clustering_two_different_methods(self, ens1): pytest.importorskip('sklearn') cluster_collection = encore.cluster( [ens1], method=[ encore.AffinityPropagation(preference=-7.5), encore.DBSCAN(min_samples=2) ]) assert len(cluster_collection[0]) == len(cluster_collection[1]), \ "Unexpected result: {0}".format(cluster_collection)
def test_clustering_two_different_methods(self): cluster_collection = encore.cluster( [self.ens1], method=[ encore.AffinityPropagation(preference=-7.5), encore.DBSCAN(min_samples=2) ]) print(cluster_collection) print(cluster_collection) assert_equal( len(cluster_collection[0]), len(cluster_collection[1]), err_msg="Unexpected result: {0}".format(cluster_collection))