Skip to content

Commit 5cf658d

Browse files
committed
Add tree interface tests
1 parent 8fba59a commit 5cf658d

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

test/reco_plan.jl

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
reco = IterativeRadonReconstructionParameters(; shape = size(images)[1:3], angles = angles, iterations = 1, reg = [L2Regularization(0.001), PositiveRegularization()], solver = CGNR);
44
algo = IterativeRadonAlgorithm(IterativeRadonParameters(pre, reco))
55

6-
6+
77
@testset "Construction" begin
88
# From algorithm
99
plan_fromAlgo = toPlan(algo)
@@ -167,6 +167,23 @@
167167
end
168168

169169
@testset "Traversal" begin
170+
plan = toPlan(algo)
171+
172+
parameter = plan.parameter
173+
@test parameter isa RecoPlan
174+
@test parameter == first(children(plan))
175+
@test plan === AbstractTrees.parent(parameter)
176+
177+
pre_plan = parameter.pre
178+
reco_plan = parameter.reco
179+
param_children = children(parameter)
180+
@test length(param_children) == 2
181+
for child in [pre_plan, reco_plan]
182+
@test child isa RecoPlan
183+
@test parameter == AbstractTrees.parent(child)
184+
@test in(child, param_children)
185+
end
186+
170187
end
171188

172189
end

0 commit comments

Comments
 (0)