Add dememoize function for restarting cache store

JuliaCollections · Jun 11, 2017 · f9758e2 · f9758e2
1 parent 1c920e0
commit f9758e2
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 1 deletion.
diff --git a/src/Memoize.jl b/src/Memoize.jl
@@ -1,5 +1,5 @@
 module Memoize
-export @memoize
+export @memoize, dememoize
 using Compat
 
 macro memoize(args...)
@@ -106,3 +106,19 @@ macro memoize(args...)
     end)
 end
 end
+
+function dememoize()
+
+    mod = current_module()
+
+    all_caches = filter( x ->
+        startswith(string(x), "##") &&
+        endswith(string(x),"_memoized_cache"),
+        names(mod, true)
+    )
+
+    for cur_cache_symbol in all_caches
+        empty!(getfield(mod, cur_cache_symbol))
+    end
+
+end
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -208,6 +208,21 @@ end
 outer()
 @test !isdefined(:inner)
 
+run = 0
+@memoize function repeat(a::Int)
+    global run += 1
+    a
+end
+@test repeat(3) == 3
+@test run == 1
+@test repeat(3) == 3
+@test run == 1
+dememoize()
+@test repeat(3) == 3
+@test run == 2
+@test repeat(3) == 3
+@test run == 2
+
 if VERSION >= v"0.5.0-dev+5235"
     @memoize function typeinf(x)
         x + 1