Skip to content

Commit a73e225

Browse files
committed
use stable rngs throughout in test/arrays.jl
1 parent 4676e02 commit a73e225

File tree

1 file changed

+14
-12
lines changed

1 file changed

+14
-12
lines changed

test/arrays.jl

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ end
4949
u = UnivariateFinite(probs, pool=missing)
5050
@test u isa UnivariateFinite
5151
@test pdf(u, "class_1") == probs[1]
52-
probs = rand(10, 2)
52+
probs = rand(rng, 10, 2)
5353
probs = probs ./ sum(probs, dims=2)
5454
u = UnivariateFinite(probs, pool=missing)
5555
@test length(u) == 10
@@ -66,9 +66,11 @@ end
6666

6767
v = categorical(1:3)
6868
@test_logs((:warn, r"Ignoring"),
69-
UnivariateFinite(v[1:2], rand(3), augment=true, pool=missing))
69+
UnivariateFinite(v[1:2], rand(rng, 3),
70+
augment=true, pool=missing))
7071
@test_logs((:warn, r"Ignoring"),
71-
UnivariateFinite(v[1:2], rand(3), augment=true, ordered=true))
72+
UnivariateFinite(v[1:2], rand(rng, 3),
73+
augment=true, ordered=true))
7274

7375
end
7476

@@ -106,7 +108,7 @@ end
106108
end
107109

108110
n = 10
109-
P = rand(n);
111+
P = rand(rng, n);
110112
all_classes = categorical(["no", "yes"], ordered=true)
111113
u = UnivariateFinite(all_classes, P, augment=true) #uni_fin_arr
112114

@@ -143,7 +145,7 @@ end
143145

144146
# check unseen probablities are a zero *array*:
145147
v = categorical(1:4)
146-
probs = rand(3)
148+
probs = rand(rng, 3)
147149
u2 = UnivariateFinite(v[1:2], probs, augment=true)
148150
@test pdf.(u2, v[3]) == zeros(3)
149151
@test isequal(logpdf.(u2, v[3]), log.(zeros(3)))
@@ -152,7 +154,7 @@ end
152154
_skip(v) = collect(skipmissing(v))
153155

154156
@testset "broadcasting: pdf.(uni_fin_arr, array_same_shape) and logpdf.(uni_fin_arr, array_same_shape)" begin
155-
v0 = categorical(rand(string.(classes(u)), n))
157+
v0 = categorical(rand(rng, string.(classes(u)), n))
156158
vm = vcat(v0[1:end-1], [missing, ])
157159
for v in [v0, vm]
158160
@test _skip(broadcast(pdf, u, v)) ==
@@ -202,8 +204,8 @@ end
202204

203205
# ordered:
204206
v = categorical(["no", "yes", "maybe", "unseen"])
205-
u1 = UnivariateFinite([v[1], v[2]], rand(5), augment=true)
206-
u2 = UnivariateFinite([v[3], v[2]], rand(6), augment=true)
207+
u1 = UnivariateFinite([v[1], v[2]], rand(rng, 5), augment=true)
208+
u2 = UnivariateFinite([v[3], v[2]], rand(rng, 6), augment=true)
207209
us = (u1, u2)
208210
u = cat(us..., dims=1)
209211
@test length(u) == length(u1) + length(u2)
@@ -222,8 +224,8 @@ end
222224

223225
# unordered:
224226
v = categorical(["no", "yes", "maybe", "unseen"], ordered=true)
225-
u1 = UnivariateFinite([v[1], v[2]], rand(5), augment=true)
226-
u2 = UnivariateFinite([v[3], v[2]], rand(6), augment=true)
227+
u1 = UnivariateFinite([v[1], v[2]], rand(rng, 5), augment=true)
228+
u2 = UnivariateFinite([v[3], v[2]], rand(rng, 6), augment=true)
227229
us = (u1, u2)
228230
u = cat(us..., dims=1)
229231
@test length(u) == length(u1) + length(u2)
@@ -252,7 +254,7 @@ end
252254
v1 = categorical(1:2, ordered=true)
253255
v2 = categorical(v1, ordered=true)
254256
levels!(v2, levels(v2) |> reverse )
255-
probs = rand(3)
257+
probs = rand(rng, 3)
256258
u1 = UnivariateFinite(v1, probs, augment=true)
257259
u2 = UnivariateFinite(v2, probs, augment=true)
258260
@test_throws DomainError vcat(u1, u2)
@@ -267,7 +269,7 @@ end
267269

268270
@testset "classes" begin
269271
v = categorical(collect("abca"), ordered=true)
270-
u1 = UnivariateFinite([v[1], v[2]], rand(5), augment=true)
272+
u1 = UnivariateFinite([v[1], v[2]], rand(rng, 5), augment=true)
271273
@test classes(u1) == collect("abc")
272274
u2 = [missing, u1...]
273275
@test classes(u2) == collect("abc")

0 commit comments

Comments
 (0)