removing prints and adding assert

rapidsai · Mar 25, 2021 · 10cb321 · 10cb321
1 parent 2ed53c1
commit 10cb321
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/python/cuml/test/test_metrics.py b/python/cuml/test/test_metrics.py
@@ -278,9 +278,7 @@ def test_silhouette_score_batched_non_monotonic():
 
     cuml_samples = cu_silhouette_samples(X=vecs, labels=labels)
     sk_samples = sk_silhouette_samples(X=vecs, labels=labels)
-    print(cuml_samples)
-    print(sk_samples)
-    array_equal(cuml_samples, sk_samples)
+    assert array_equal(cuml_samples, sk_samples)
 
     vecs = np.array([[0.0, 0.0, 0.0], [1.0, 1.0, 1.0], [10.0, 10.0, 10.0]])
     labels = np.array([1, 1, 3])
@@ -289,7 +287,7 @@ def test_silhouette_score_batched_non_monotonic():
     sk_samples = sk_silhouette_samples(X=vecs, labels=labels)
     print(cuml_samples)
     print(sk_samples)
-    array_equal(cuml_samples, sk_samples)
+    assert array_equal(cuml_samples, sk_samples)
 
 
 def score_homogeneity(ground_truth, predictions, use_handle):