Disable GridPopulation with SurvivalQuantities (#311)

Genentech · May 8, 2024 · 4aa31f9 · 4aa31f9
1 parent 4c63b98
commit 4aa31f9
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/R/SurvivalQuantities.R b/R/SurvivalQuantities.R
@@ -71,6 +71,10 @@ SurvivalQuantities <- function(
     type <- match.arg(type)
     assert_class(object, "JointModelSamples")
     assert_class(grid, "Grid")
+    assert_that(
+        !is(grid, "GridPopulation"),
+        msg = "GridPopulation objects are not supported for `SurvivalQuantities`"
+    )
 
     time_grid <- seq(
         from = 0,

diff --git a/tests/testthat/test-Grid.R b/tests/testthat/test-Grid.R
@@ -696,3 +696,15 @@ test_that("GridPopulation() works as expected for Longitudinal models", {
     expect_equal(actual$group, expected$group)
 
 })
+
+test_that("GridPopulation() doesn't work with SurvivalQuantities", {
+    expect_error(
+        SurvivalQuantities(
+            fixtures_gsf$mp,
+            grid = GridPopulation(
+                times = seq(1, 4, by = 0.02)
+            )
+        ),
+        regex = "not supported"
+    )
+})