test: test selection

src/bin/enc.rs

@@ -196,8 +196,20 @@ fn encrypted_knn_classify(
     }
     println!(); // New line after progress bar
 
-    println!("📊 Sorting {} distances...", distances.len());
-    bitonic_sort_encrypted(&mut distances, true);
+    println!("📊 Finding {} smallest distances using selection...", k);
+    
+    // Use selection instead of full sorting to match plain version behavior
+    for target_pos in 0..k.min(distances.len()) {
+        // Find minimum in remaining elements [target_pos..]
+        for i in (target_pos + 1)..distances.len() {
+            let should_swap = distances[i].distance.lt(&distances[target_pos].distance);
+            
+            // Use split_at_mut to avoid borrowing issues
+            let (left, right) = distances.split_at_mut(i);
+            encrypted_conditional_swap(&mut left[target_pos], &mut right[0], &should_swap);
+        }
+    }
+    
     distances.truncate(k);
 
     distances.iter().map(|n| n.index.clone()).collect()