From 1356bb746e5bc4bf7f55fd5c6faa2901f6b2726c Mon Sep 17 00:00:00 2001
From: Patrick Pelissier <patrick.pelissier@gmail.com>
Date: Sat, 6 Jul 2019 10:44:49 +0200
Subject: [PATCH] Add more tests

---
 tests/test-mbptree.c | 45 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/tests/test-mbptree.c b/tests/test-mbptree.c
index 97040a42..e849520a 100644
--- a/tests/test-mbptree.c
+++ b/tests/test-mbptree.c
@@ -195,6 +195,51 @@ static void test3(void)
   }
   assert(btree_size(b) == 0);
 
+  const int prime[5] = { 3, 5, 11, 13, 17};
+  const int prime_size = 5;
+  for(int k = 0; k < prime_size; k++) {
+    int p = prime[k];
+    for(int i = 0; i < 1024; i++) {
+      btree_set_at(b, (i*p) % 1024, i);
+      assert(*btree_cget(b, (i*p) % 1024) == i);
+    }
+    assert(btree_size(b) == 1024);
+    for(int i = 0; i < 1024; i++) {
+      assert(*btree_cget(b, (i*p) % 1024) == i);
+    }
+    for(int i = 0; i < 1024; i++) {
+      bool r = btree_erase(b, (i*p) % 1024);
+      assert(r);
+      assert(btree_cget(b, (i*p) % 1024) == NULL);
+    }
+    assert(btree_size(b) == 0);
+  }
+
+  for(int k = 0; k < prime_size; k++) {
+    int p = prime[k];
+    for(int i = 0; i < 512; i++) {
+      btree_set_at(b, (i*p) % 1024, i);
+      assert(*btree_cget(b, (i*p) % 1024) == i);
+    }
+    assert(btree_size(b) == 512);
+    for(int i = 256; i < 512; i++) {
+      bool r = btree_erase(b, (i*p) % 1024);
+      assert(r);
+      assert(btree_cget(b, (i*p) % 1024) == NULL);
+    }
+    for(int i = 256; i < 1024; i++) {
+      btree_set_at(b, (i*p) % 1024, i);
+      assert(*btree_cget(b, (i*p) % 1024) == i);
+    }
+    assert(btree_size(b) == 1024);
+    for(int i = 0; i < 1024; i++) {
+      bool r = btree_erase(b, (i*p) % 1024);
+      assert(r);
+      assert(btree_cget(b, (i*p) % 1024) == NULL);
+    }
+    assert(btree_size(b) == 0);
+  }
+  
   btree_clear(b);
 }