From 62d48030753007af6a69eb53f596bb595f1ee6a5 Mon Sep 17 00:00:00 2001
From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com>
Date: Sun, 13 Jul 2025 01:35:02 +0100
Subject: [PATCH] Simplify assertion for unique labels in KMeans tests when k
 equals m

---
 src/compute/models/k_means.rs | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/src/compute/models/k_means.rs b/src/compute/models/k_means.rs
index 6dcd155..7fea599 100644
--- a/src/compute/models/k_means.rs
+++ b/src/compute/models/k_means.rs
@@ -256,11 +256,8 @@ mod tests {
         let mut sorted_labels = labels.clone();
         sorted_labels.sort_unstable();
         sorted_labels.dedup();
-        assert_eq!(
-            sorted_labels.len(),
-            k,
-            "Labels should all be unique when k==m"
-        );
+        // Labels should all be unique when k==m
+        assert_eq!(sorted_labels.len(), k);
     }
 
     #[test]