fix time zero when given distribution of deaths

JuliaActuary · Jun 4, 2020 · 939ebbe · 939ebbe
1 parent f0a1ef9
commit 939ebbe
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/death_distribution.jl b/src/death_distribution.jl
@@ -103,8 +103,10 @@ A function to calculate non-whole year survivorship, where you must
     mortaility applies through the rest of the year.
 """
 function p(tbl, issue_age, duration, time, dist::Uniform)
-
-    if time > 1
+    if time == 0 
+        return 1.0
+
+    elseif time > 1
         whole_time = trunc(Int, time)
         p′ = p(tbl, issue_age, duration, whole_time)
         return p′ *

diff --git a/test/distribution.jl b/test/distribution.jl
@@ -27,6 +27,10 @@
                       round(1 - target[i], digits = 4)
             end
         end
+
+        # test time zero when given distribution of deaths
+        @test q(soa_mort,0,1,0,Uniform()) == 0.0
+        @test p(soa_mort,0,1,0,Uniform()) == 1.0
     end
     @testset "Multi-year examples" begin
         mort = UltimateMortality([0.20, 0.50])