diff --git a/tests/test_memoize.py b/tests/test_memoize.py index 9b885d0fd7d2c739fddd1d7bc15bb8eb753f5ba0..386674659b4a9af2648dadf445efd1a84a170f37 100644 --- a/tests/test_memoize.py +++ b/tests/test_memoize.py @@ -242,7 +242,29 @@ def test_skipUnchanged(): wrapped('key4', np.zeros((1, 4))) assert timesCalled['key4'] == 2 + timesCalled['key1'] = 0 + timesCalled['key2'] = 0 + timesCalled['key3'] = 0 + wrapped('key1', 1) + wrapped('key2', 2) + wrapped('key3', 3) + assert timesCalled['key1'] == 1 + assert timesCalled['key2'] == 1 + assert timesCalled['key3'] == 1 + wrapped('key1', 1) + wrapped('key2', 2) + wrapped('key3', 3) + assert timesCalled['key1'] == 1 + assert timesCalled['key2'] == 1 + assert timesCalled['key3'] == 1 + wrapped.invalidate() + wrapped('key1', 1) + wrapped('key2', 2) + wrapped('key3', 3) + assert timesCalled['key1'] == 2 + assert timesCalled['key2'] == 2 + assert timesCalled['key3'] == 2 def test_Instanceify():