From b0a1ddd80531325bc2e1c1fb2be276f81e60f0ad Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Paul=20J=C3=BCr=C3=9F?= <p.juerss@uke.de>
Date: Wed, 28 Feb 2024 14:41:31 +0100
Subject: [PATCH] Add tests for `Vessel.jl`

---
 test/runtests.jl | 72 +++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 59 insertions(+), 13 deletions(-)

diff --git a/test/runtests.jl b/test/runtests.jl
index 104dd5d..dd86a5f 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -4,21 +4,67 @@ using Test
 
 @testset "TrainingPhantoms.jl" begin
   N = (40,40,40)
+  @testset "Vessel" begin    
+    @testset "getDiameterRoute" begin
+      route = [1, 2, 3, 4, 5]
+      diameter = 2.0
+      change_diameter_splitting = 0.5
+      splitnr = 2
+  
+      @testset "Normal operation" begin
+          result = TrainingPhantoms.getDiameterRoute(route, diameter, change_diameter_splitting, splitnr)
+          @test length(result) == length(route)
+          @test result[1] == (1/change_diameter_splitting)*diameter
+          @test result[end] == diameter
+      end
+  
+      @testset "No split" begin
+          result = TrainingPhantoms.getDiameterRoute(route, diameter, change_diameter_splitting, 1)
+          @test all(result .== diameter)
+      end
+    end
+  end
 
-  # Vessel Phantom
-  im = vesselPhantom(N; start=(1, 20, 20), angle_xy=0.0, angle_xz=0.0, 
-                                diameter=2, split_prob=0.5, change_prob=0.5, max_change=0.2, splitnr=1,
-                                rng = StableRNG(1));
+  @testset "vesselPath Tests" begin
+    rng = StableRNG(1)
+    @testset "Normal operation" begin
+        route, diameter_route = TrainingPhantoms.vesselPath(N; start=(1,20,20), angle_xy=0.0, angle_xz=0.0, diameter=2, 
+                                           split_prob=0.5, change_prob=0.5, max_change=0.2, 
+                                           splitnr=1, max_number_splits=2, stepsize=0.25, 
+                                           change_diameter_splitting=4/5, split_prob_factor=0.5, 
+                                           change_prob_increase=0.01, rng=rng)
+        @test length(route) == length(diameter_route)
+    end
+    @testset "Start outside volume" begin
+        route, diameter_route = TrainingPhantoms.vesselPath(N; start=(1,50,20), angle_xy=0.0, angle_xz=0.0, diameter=2, 
+                                           split_prob=0.5, change_prob=0.5, max_change=0.2, 
+                                           splitnr=1, max_number_splits=2, stepsize=0.25, 
+                                           change_diameter_splitting=4/5, split_prob_factor=0.5, 
+                                           change_prob_increase=0.01, rng=rng)
+        @test length(route) == 1
+        @test length(diameter_route) == 1
+    end
+  end
+  @testset "vesselPhantom" begin
+    im = vesselPhantom(N; start=(1, 20, 20), angle_xy=0.0, angle_xz=0.0, 
+                       diameter=2, split_prob=0.5, change_prob=0.5, max_change=0.2, splitnr=1,
+                       rng = StableRNG(1));
 
-  im2 = vesselPhantom(N; start=(1, 20, 20), angle_xy=0.0, angle_xz=0.0, 
-                                diameter=2, split_prob=0.5, change_prob=0.5, max_change=0.2, splitnr=1,
-                                rng = StableRNG(1));
-  @test im ≈ im2
+    im2 = vesselPhantom(N; start=(1, 20, 20), angle_xy=0.0, angle_xz=0.0, 
+                        diameter=2, split_prob=0.5, change_prob=0.5, max_change=0.2, splitnr=1,
+                        rng = StableRNG(1));
+    @test im ≈ im2
+    @test size(im) == N
+    @test maximum(im) == 1.0
+    @test minimum(im) == 0.0
+  end
 
-   # Ellipsoid Phantom
-  im = ellipsoidPhantom(N; rng=StableRNG(1))
-  im2 = ellipsoidPhantom(N; rng=StableRNG(1))
-  @test im ≈ im2
 
-  #isosurface(im, isovalue=0.2, rotation=110)
+  @testset "Ellipsoid" begin
+    # Ellipsoid Phantom
+    im = ellipsoidPhantom(N; rng=StableRNG(1))
+    im2 = ellipsoidPhantom(N; rng=StableRNG(1))
+    @test im ≈ im2
+    @test size(im) == N
+  end
 end