Improve tests

!test

Author: awolverp

python/cachebox/_cachebox.py

@@ -1939,7 +1939,7 @@ def popitem(self) -> typing.Tuple[KT, VT]:
         """
         try:
             val = self._raw.popitem()
-        except _core.CoreKeyError:
+        except _core.CoreKeyError: # pragma: no cover
             raise KeyError() from None
         else:
             return val.pack2()

python/cachebox/_core.pyi

@@ -68,3 +68,6 @@ class BaseCacheImpl(typing.Generic[KT, VT]):
     def keys(self) -> typing.Iterable[KT]: ...
     def values(self) -> typing.Iterable[VT]: ...
     def items(self) -> typing.Iterable[typing.Tuple[KT, VT]]: ...
+    def __copy__(self) -> "BaseCacheImpl[KT, VT]": ...
+    def __deepcopy__(self, memo) -> "BaseCacheImpl[KT, VT]": ...
+    def copy(self) -> "BaseCacheImpl[KT, VT]": ...

python/tests/mixin.py

@@ -431,3 +431,32 @@ def _test_pickle(self, check_order: typing.Callable):
             assert c1 == c2
             assert c1.capacity() == c2.capacity()
             check_order(c1, c2)
+
+    def test_copy(self):
+        import copy
+
+        # shallow copy
+        c1 = self.CACHE(maxsize=0, **self.KWARGS)
+        c1.insert('dict', {})
+        c2 = c1.copy()
+
+        assert c2 == c1
+        c2['dict'][1] = 1
+
+        assert c1['dict'][1] == 1
+
+        c2.insert(1, 1)
+        assert 1 not in c1
+
+        # deepcopy
+        c1 = self.CACHE(maxsize=0, **self.KWARGS)
+        c1.insert('dict', {})
+        c2 = copy.deepcopy(c1)
+
+        assert c2 == c1
+        c2['dict'][1] = 1
+
+        assert 1 not in c1['dict']
+
+        c2.insert(1, 1)
+        assert 1 not in c1

python/tests/test_caches.py

@@ -430,6 +430,15 @@ def test_popitem_with_expire(self):
         with pytest.raises(KeyError):
             obj.popitem_with_expire()
 
+    def test_items_with_expire(self):
+        # no need to test completely items_with_expire
+        # because it's tested in test_iterators
+        obj = TTLCache(10, 3, {1: 2, 3: 4})
+        for key, val, ttl in obj.items_with_expire():
+            assert key in obj
+            assert val == obj[key]
+            assert isinstance(ttl, float)
+
 
 class TestVTTLCache(_TestMixin):
     CACHE = VTTLCache
@@ -571,5 +580,14 @@ def inner(c1, c2):
         c2 = pickle.loads(pickle.dumps(c1))
 
         assert len(c2) == len(c1)
-        assert c1.capacity() == c2.capacity()
+        assert abs(c2.capacity() - c1.capacity()) < 2
         inner(c1, c2)
+
+    def test_items_with_expire(self):
+        # no need to test completely items_with_expire
+        # because it's tested in test_iterators
+        obj = VTTLCache(10, {1: 2, 3: 4}, ttl=10)
+        for key, val, ttl in obj.items_with_expire():
+            assert key in obj
+            assert val == obj[key]
+            assert isinstance(ttl, float)