Fix flaky unit tests (#288)

Author: jparismorgan

apis/python/test/test_type_erased_module.py

@@ -272,7 +272,7 @@ def test_inplace_build_infinite_query_IndexIVFFlat():
         if nprobe == 8:
             assert recall > 0.925
         if nprobe == 32:
-            assert recall == 1.0
+            assert recall >= 0.999
 
 
 def test_read_index_and_infinite_query():

src/include/test/unit_api_ivf_flat_index.cc

@@ -201,7 +201,7 @@ TEST_CASE(
     auto nt = num_vectors(t);
     auto recall = ((double)intersections) / ((double)nt * k_nn);
     if (nprobe == 32) {
-      CHECK(std::abs(recall - 1.0) <= 1e-3);
+      CHECK(recall >= 0.999);
     } else if (nprobe == 8) {
       CHECK(recall > 0.925);
     }
@@ -259,7 +259,7 @@ TEST_CASE(
     CHECK(nt == nv);
     auto recall = ((double)intersections_a) / ((double)nt * k_nn);
     if (nprobe == 32) {
-      CHECK(recall == 1.0);
+      CHECK(recall >= 0.999);
     } else if (nprobe == 8) {
       CHECK(recall > 0.925);
     }
@@ -305,7 +305,7 @@ TEST_CASE(
     CHECK(nt == nv);
     auto recall = ((double)intersections_a) / ((double)nt * k_nn);
     if (nprobe == 32) {
-      CHECK(std::abs(recall - 1.0) <= 1e-3);
+      CHECK(recall >= 0.999);
     } else if (nprobe == 8) {
       CHECK(recall > 0.925);
     }

src/include/test/unit_logging.cc

@@ -170,7 +170,7 @@ TEST_CASE("logging: ordering", "[logging]") {
 
   std::cout << f_t << " " << g_t << " " << h_t << " " << i_t << std::endl;
 
-  CHECK((i_t > 799 && i_t < 880));
+  CHECK((i_t > 799 && i_t < 890));
   CHECK((h_t > 499 && h_t < 560));
   CHECK((g_t > 499 && g_t < 560));
   CHECK((f_t > 499 && f_t < 560));