Skip to content

Commit e5064c1

Browse files
committed
test: fix issue where test eval fails due to incomplete eval
1 parent 7badcc1 commit e5064c1

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

test/test_buffered_evaluation.jl

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -131,8 +131,9 @@ end
131131

132132
for turbo in (false, true), i in 1:100
133133
# Generate a random tree with varying size (1-10 nodes)
134-
n_nodes = rand(1:10)
135-
tree = gen_random_tree_fixed_size(n_nodes, operators, size(X, 1), Float64, Node)
134+
rng = Random.MersenneTwister(i)
135+
n_nodes = rand(rng, 1:10)
136+
tree = gen_random_tree_fixed_size(n_nodes, operators, size(X, 1), Float64, Node, rng)
136137

137138
# Regular evaluation
138139
eval_options_no_buffer = EvalOptions(; turbo)
@@ -142,12 +143,12 @@ end
142143

143144
# Buffer evaluation
144145
buffer = Array{Float64}(undef, 2n_nodes, size(X, 2))
145-
buffer_ref = Ref(rand(1:10)) # Random starting index (will be reset)
146+
buffer_ref = Ref(rand(rng, 1:10)) # Random starting index (will be reset)
146147
eval_options = EvalOptions(; turbo, buffer=ArrayBuffer(buffer, buffer_ref))
147148
result2, ok2 = eval_tree_array(tree, X, operators; eval_options)
148149

149150
# Results should be identical
150-
@test isapprox(result1, result2; atol=1e-10)
151+
@test isapprox(result1, result2; atol=1e-10) || (!ok1 && !ok2)
151152
@test ok1 == ok2
152153
end
153154
end

0 commit comments

Comments
 (0)