change fakedata rng in tests

t-bltg · Jul 23, 2021 · cc1a328 · cc1a328
1 parent cd06648
commit cc1a328
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/utils.jl b/src/utils.jl
@@ -303,7 +303,9 @@ function convert_to_polar(theta, r, r_extrema = ignorenan_extrema(r))
     x, y
 end
 
-function fakedata(sz...; rng=Random.GLOBAL_RNG(PLOTS_SEED))
+fakedata(sz::Int...) = fakedata(Random.seed!(PLOTS_SEED), sz...)
+
+function fakedata(rng::AbstractRNG, sz...)
     y = zeros(sz...)
     for r in 2:size(y,1)
         y[r,:] = 0.95 * vec(y[r-1,:]) + randn(rng, size(y,2))

diff --git a/test/imgcomp.jl b/test/imgcomp.jl
@@ -6,7 +6,7 @@ function replace_rand!(ex::Expr)
    for arg in ex.args
        replace_rand!(arg)
    end
-   if ex.head === :call && ex.args[1] ∈ (:rand, :randn)
+   if ex.head === :call && ex.args[1] ∈ (:rand, :randn, :(Plots.fakedata))
        pushfirst!(ex.args, ex.args[1])
        ex.args[2] = :rng
    end
@@ -41,6 +41,7 @@ function image_comparison_tests(
         expr = Expr(:block)
         append!(expr.args, example.exprs)
         fix_rand!(expr)
+        @debug expr
         eval(expr)
         png(fn)
     end