unregister cache in doctest

dask · Aug 3, 2015 · dcfebeb · dcfebeb
1 parent 5f507bc
commit dcfebeb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/dask/diagnostics/cache.py b/dask/diagnostics/cache.py
@@ -16,6 +16,7 @@ class Cache(Diagnostic):
     ...     result = x.compute()  # doctest: +SKIP
 
     >>> cache.register()    # or use globally
+    >>> cache.unregister()
     """
 
     def __init__(self, cache, *args, **kwargs):

diff --git a/dask/diagnostics/tests/test_cache.py b/dask/diagnostics/tests/test_cache.py
@@ -2,6 +2,7 @@
 import cachey
 from dask.threaded import get
 from operator import add
+from dask.context import _globals
 
 
 flag = []
@@ -28,6 +29,8 @@ def test_cache():
 
     assert flag == [2]  # no x present
 
+    assert not _globals['callbacks']
+
 
 def test_cache_with_number():
     c = Cache(10000, limit=1)