add t > 0 enforcement for lower incomplete gamma

epinowcast · Oct 8, 2024 · 79a12aa · 79a12aa
1 parent 95073b5
commit 79a12aa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/R/primary_censored_dist.R b/R/primary_censored_dist.R
@@ -283,11 +283,9 @@ primary_censored_cdf.pcens_pweibull_dunif <- function(
 
   g <- function(t) {
     # Use the lower incomplete gamma function
+    t <- pmax(t, 0)
     scaled_t <- (t * inv_scale)^shape
     vapply(scaled_t, function(x) {
-      if (x <= 0) {
-        return(0)
-      }
       pracma::gammainc(x, 1 + inv_shape)["lowinc"]
     }, numeric(1))
   }

diff --git a/tests/testthat/test-primary_censored_dist.R b/tests/testthat/test-primary_censored_dist.R
@@ -258,8 +258,8 @@ test_that(
     dprimary <- dunif
 
     shapes <- c(0.5, 1, 2, 3)
-    scales <- c(0.5, 1, 2, 5)
-    pwindows <- c(1, 2, 5, 10)
+    scales <- c(0.5, 1, 2)
+    pwindows <- c(1, 2, 5)
 
     for (shape in shapes) {
       for (scale in scales) {