remove missing values in test()

qddyy · Nov 27, 2024 · d2ac38a · d2ac38a
1 parent 9def687
commit d2ac38a
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/R/PermuTest.R b/R/PermuTest.R
@@ -443,7 +443,8 @@ PermuTest <- R6Class(
 )
 
 get_data <- function(call, env) {
-    data_exprs <- as.list(call)[-1]
+    data_exprs <- as.list.default(call)[-1]
+
     n_data <- length(data_exprs)
 
     if (n_data == 1 && is.list(data_1 <- eval(data_exprs[[1]], envir = env))) {
@@ -465,11 +466,19 @@ get_data <- function(call, env) {
             }
 
             data_i <- eval(data_exprs[[i]], envir = env)
+
             if (!is.numeric(data_i)) {
                 stop("Sample ", i, " is not numeric")
-            } else if (anyNA(data_i)) {
-                stop("Sample ", i, " contains NA")
-            } else data_i
+            }
+            if (anyNA(data_i)) {
+                warning("Sample ", i, " contains missing values, removed")
+                data_i <- data_i[!is.na(data_i)]
+            }
+            if (length(data_i) < 1) {
+                stop("Sample ", i, " does not contain enough observations")
+            }
+
+            data_i
         }
     ), data_names)
 }