Add tests for ordered weighted sampling with replacement (#59)

JuliaDynamics · Apr 17, 2024 · 81d91ea · 81d91ea
1 parent c41ae92
commit 81d91ea
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/weighted_sampling_multi_tests.jl b/test/weighted_sampling_multi_tests.jl
@@ -26,7 +26,8 @@ function prob_no_replace(k)
 end
 
 @testset "Weighted sampling multi tests" begin
-    combs = Iterators.product([(algAExpJ, algARes, algWRSWRSKIP), (false, )]...)
+    combs = vec(collect(Iterators.product([(algAExpJ, algARes, algWRSWRSKIP), (false, )]...)))
+    push!(combs, (algWRSWRSKIP, true))
     @testset "method=$method ordered=$ordered" for (method, ordered) in combs
         a, b = 1, 10
         # test return values of iter with known lengths are inrange
@@ -67,7 +68,7 @@ end
         @test ordered ? issorted(s) : true
 
         weight2(el) = el <= 5 ? 1.0 : 2.0
-        rng = StableRNG(43)
+        rng = StableRNG(41)
         iters = (a:b, Iterators.filter(x -> x != b+1, a:b+1))
         sizes = (1, 2)
         for it in iters