survival POC

[skip CI]
DanChaltiel · Aug 24, 2024 · 46bbc4e · 46bbc4e
1 parent 9253ed1
commit 46bbc4e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/testthat/test-quick_surv.R b/tests/testthat/test-quick_surv.R
@@ -22,12 +22,12 @@ test_that("survival POC", {
     fit = survival::survfit(surv~group, data=df)
     times = sort(fit$time)
 
-    a = summary(fit, times=times, extend=TRUE, data.frame=TRUE)
+    x = summary(fit, times=times, extend=TRUE)
+    a = data.frame(strata=x$strata, time=x$time, n.risk=x$n.risk, n.event=x$n.event, surv=x$surv)
+
     a %>%
       dplyr::filter(strata=="group=B") %>%
-      dplyr::filter(time>120 & time<150) %>%
-      dplyr::select(strata, time, n.risk, n.event, surv)
-
+      dplyr::filter(time>120 & time<150)
   })