update wlr.counting_process following the unsuccessful test in `tes…

…t-unvalidated_wlr`

R/counting_process.R

@@ -148,7 +148,14 @@ counting_process <- function(x, arm) {
   setDF(ans)
   class(ans) <- c("counting_process", class(ans))
 
-  attr(ans, "ratio") <- attributes(x)$ratio
+  # if `x` is generated by sim_pw_surv
+  if ("generate_by_simpwsurv" %in% names(attributes(x))) {
+    ratio <- attributes(x)$ratio
+  } else {
+  # if not, calcualte the emperical ratio
+    ratio <- sum(x$treatment == arm) / sum(x$treatment != arm)
+  }
+  attr(ans, "ratio") <- ratio
 
   return(ans)
 }

R/wlr.R

@@ -114,6 +114,10 @@
 #' # Users can specify the randomization ratio to calculate the statistical information under H0
 #' x |> wlr(weight = fh(rho = 0, gamma = 0.5), ratio = 2)
 #'
+#' x |>
+#'   counting_process(arm = "experimental") |>
+#'   wlr(weight = fh(rho = 0, gamma = 0.5), ratio = 2)
+#'
 #' # If users don't provide the randomization ratio, we will calculate the emperical ratio
 #' x |> wlr(weight = fh(rho = 0, gamma = 0.5))
 #'
@@ -129,7 +133,7 @@ wlr <- function(data, weight, return_variance = FALSE, ratio = NULL) {
 wlr.tte_data <- function(data, weight, return_variance = FALSE, ratio = NULL) {
   # if the `data` is NOT generated by sim_pw_surv
   # - if user input the randomization ratio, we will directly take its values
-  # - otherwise, we calculate the emperical ratio
+  # - otherwise, we calculate the empirical ratio
   if (!"generate_by_simpwsurv" %in% names(attributes(data))) {
     if (is.null(ratio)) {
       ratio <- sum(data$treatment == "experimental") / sum(data$treatment == "control")
@@ -148,6 +152,9 @@ wlr.tte_data <- function(data, weight, return_variance = FALSE, ratio = NULL) {
 wlr.counting_process <- function(data, weight, return_variance = FALSE, ratio = NULL) {
   x <- data
 
+  if (is.null(ratio)) {
+    ratio <- attributes(data)$ratio
+  }
   q_e <- ratio / (1 + ratio)
   q_c <- 1 - q_e