add tests to check if key exists

btjm123 · Aug 9, 2024 · c3f6e9d · c3f6e9d
1 parent 9883cdf
commit c3f6e9d
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/tests/ChainingHashMap.cc b/tests/ChainingHashMap.cc
@@ -88,4 +88,17 @@ TEST(ChainingHashMap, Resize)
     map[7] = 70;
     map[8] = 80;
     EXPECT_EQ(map.getBucketCount(), 16);
+}
+
+// Test for existence of key
+TEST(ChainingHashMap, ExistenceOfKey)
+{
+    benn::ChainingHashMap<int, int> map;
+
+    EXPECT_EQ(map.count(0), 0);
+    map[0] = 10;
+
+    EXPECT_EQ(map.count(0), 1);
+    EXPECT_EQ(map.count(1), 0);
+    EXPECT_EQ(map.count(10), 0);
 }
diff --git a/tests/OpenAddressingHashMap.cc b/tests/OpenAddressingHashMap.cc
@@ -88,4 +88,17 @@ TEST(OpenAddressingHashMap, Resize)
     map[7] = 70;
     map[8] = 80;
     EXPECT_EQ(map.getBucketCount(), 16);
+}
+
+// Test for existence of key
+TEST(OpenAddressingHashMap, ExistenceOfKey)
+{
+    benn::OpenAddressingHashMap<int, int> map;
+
+    EXPECT_EQ(map.count(0), 0);
+    map[0] = 10;
+
+    EXPECT_EQ(map.count(0), 1);
+    EXPECT_EQ(map.count(1), 0);
+    EXPECT_EQ(map.count(10), 0);
 }