test(test_core): add tests for many* methods

Author: Rizhiy

class_cache/core.py

@@ -75,6 +75,8 @@ def __delitem__(self, key: KeyType) -> None:
         if key in self._data:
             del self._lru_queue[key]
         self._data.pop(key, None)
+        if key in self._to_write:
+            self._to_write.remove(key)
         self._to_delete.add(key)
 
     def write(self) -> None:

tests/test_core.py

@@ -24,6 +24,15 @@ def get_new_cache(id_: str = None, *, clear=True, **kwargs) -> Cache:
     return cache
 
 
+def get_cache_with_items(*args, **kwargs) -> tuple[Cache, dict[int, str]]:
+    cache = get_new_cache(*args, **kwargs)
+
+    items = {i: str(i) for i in range(16)}
+
+    cache.set_many(items.items())
+    return cache, items
+
+
 def test_basic_cache(test_key, test_value):
     first = get_new_cache()
     first[test_key] = test_value
@@ -142,4 +151,49 @@ def test_max_memory_usage():
     assert end_max_memory_usage - starting_max_memory < 3_000
 
 
+def test_equals():
+    cache = get_new_cache()
+    cache2 = get_new_cache(max_items=64)
+    assert cache == cache2
+
+    cache3 = get_new_cache("Another")
+    assert cache != cache3
+
+
+def test_contains_many():
+    cache, items = get_cache_with_items()
+
+    keys = tuple(items.keys())
+    in_keys, ins = zip(*cache.contains_many(keys))
+
+    assert in_keys == keys
+    assert all(ins)
+
+
+def test_get_many():
+    cache, items = get_cache_with_items()
+
+    keys = tuple(items.keys())
+    in_keys, values = zip(*cache.get_many(keys))
+
+    assert in_keys == keys
+    assert values == tuple(items.values())
+
+
+def test_set_many():
+    cache, items = get_cache_with_items()
+
+    for key, value in items.items():
+        assert value == cache[key]
+
+
+def test_del_many():
+    cache, items = get_cache_with_items()
+
+    keys = tuple(items.keys())
+    cache.del_many(keys)
+
+    assert len(cache) == 0
+
+
 # TODO: Add parallel test for cache as well (threading)