src/compute/models/k_means.rs

@@ -256,8 +256,11 @@ mod tests {
         let mut sorted_labels = labels.clone();
         sorted_labels.sort_unstable();
         sorted_labels.dedup();
-        // Labels should all be unique when k==m
-        assert_eq!(sorted_labels.len(), k);
+        assert_eq!(
+            sorted_labels.len(),
+            k,
+            "Labels should all be unique when k==m"
+        );
     }
 
     #[test]
@@ -321,5 +324,4 @@ mod tests {
         assert_eq!(predicted_label.len(), 1);
         assert!(predicted_label[0] < k);
     }
-
 }