clean up normalize_function_limitied_size test

mrocklin · Mar 15, 2017 · 066e6f7 · 066e6f7
1 parent b64169f
commit 066e6f7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dask/base.py b/dask/base.py
@@ -269,7 +269,7 @@ def normalize_function(func):
         return function_cache[func]
     except KeyError:
         result = _normalize_function(func)
-        if len(function_cache) > 500:  # clear half of cache if full
+        if len(function_cache) >= 500:  # clear half of cache if full
             for k in list(function_cache)[::2]:
                 del function_cache[k]
         function_cache[func] = result

diff --git a/dask/tests/test_base.py b/dask/tests/test_base.py
@@ -457,4 +457,4 @@ def test_normalize_function_limited_size():
     for i in range(1000):
         normalize_function(lambda x: x)
 
-    assert 50 < len(function_cache) < 500
+    assert 50 < len(function_cache) < 600