Merge branch 'u/arrayka/diskann-disk-tests3' of https://github.com/microsoft/DiskANN into u/arrayka/diskann-disk-tests3

Author: Alex Razumov (from Dev Box)

diskann-disk/src/build/chunking/continuation/utils.rs

@@ -208,7 +208,7 @@ mod tests {
             let mut yielded = self.yielded.lock().unwrap();
             if !*yielded {
                 *yielded = true;
-                ContinuationGrant::Yield(std::time::Duration::from_millis(1))
+                ContinuationGrant::Yield(std::time::Duration::ZERO)
             } else {
                 // After yielding once, always continue
                 ContinuationGrant::Continue

diskann-disk/src/search/pq/pq_scratch.rs

@@ -137,8 +137,8 @@ mod tests {
         // Query shorter than dim should fail
         let short_query: Vec<f32> = (1..dim).map(|i| i as f32).collect(); // dim-1 elements
         let err = pq_scratch.set(&short_query).unwrap_err();
+        assert_eq!(err.kind(), diskann::ANNErrorKind::DimensionMismatchError);
         assert!(err.to_string().contains("expected query of length"));
-    }
 
     #[test]
     fn test_pq_scratch_set_accepts_oversized_query() {

diskann-disk/src/storage/quant/generator.rs

@@ -627,11 +627,11 @@ mod generator_tests {
         )
         .unwrap();
 
-        let err = generator.validate_params(num_points, &storage_provider);
-        assert!(err.is_err());
-        let err_msg = format!("{:?}", err.unwrap_err());
-        assert!(err_msg.contains("expected compressed file"));
-
+        let err = generator
+            .validate_params(num_points, &storage_provider)
+            .unwrap_err();
+        assert_eq!(err.kind(), diskann::ANNErrorKind::FileNotFoundError);
+        assert!(err.to_string().contains("expected compressed file"));
         Ok(())
     }
 }