Fix bug in nda_mem test

TRIQS · Aug 12, 2024 · d00f18d · d00f18d
1 parent 16acec3
commit d00f18d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/test/c++/nda_mem.cpp b/test/c++/nda_mem.cpp
@@ -297,14 +297,16 @@ TEST(NDA, MemoryMultiBucketAllocator) {
   bucket2[0] = allo.allocate_zero(chunksize);
   EXPECT_EQ(allo.buckets().size(), 2);
   EXPECT_TRUE(allo.owns(bucket2[0]));
-  EXPECT_TRUE(allo.buckets()[1].owns(bucket2[0]));
+  EXPECT_TRUE(allo.buckets()[0].owns(bucket2[0]) != allo.buckets()[1].owns(bucket2[0]));
 
   // deallocate and reallocate in the 1st bucket
+  std::size_t idx = 0;
+  if (allo.buckets()[1].owns(bucket1[20])) idx = 1;
   allo.deallocate(bucket1[20]);
-  EXPECT_FALSE(allo.buckets()[0].is_full());
+  EXPECT_FALSE(allo.buckets()[idx].is_full());
   auto mb_realloc = allo.allocate_zero(chunksize);
-  EXPECT_TRUE(allo.buckets()[0].owns(mb_realloc));
-  EXPECT_TRUE(allo.buckets()[0].is_full());
+  EXPECT_TRUE(allo.buckets()[idx].owns(mb_realloc));
+  EXPECT_TRUE(allo.buckets()[idx].is_full());
 
   // erase 2nd bucket
   allo.deallocate(bucket2[0]);