Test memoize.skip()

Suor · Jun 23, 2024 · 207a781 · 207a781
1 parent 859056d
commit 207a781
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/test_calc.py b/tests/test_calc.py
@@ -43,6 +43,25 @@ def mul(x, by=1):
     assert calls == [(0, 1), (1, 1), (0, 1), (1, 1)]
 
 
+def test_memoize_skip():
+    @memoize
+    def inc(x):
+        calls.append(x)
+        if x == 2:
+            raise memoize.skip
+        if x == 3:
+            raise memoize.skip(42)
+        return x + 1
+
+    calls = []
+    assert inc(1) == 2
+    assert inc(2) is None
+    assert inc(2) is None
+    assert inc(3) == 42
+    assert inc(3) == 42
+    assert calls == [1, 2, 2, 3, 3]
+
+
 def test_memoize_memory():
     @memoize
     def inc(x):