refactor: use modulo

r-lib · Feb 24, 2025 · 37aa592 · 37aa592
1 parent 48b44af
commit 37aa592
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/R/mock2-helpers.R b/R/mock2-helpers.R
@@ -41,7 +41,13 @@ mock_output_sequence <- function(values, recycle = FALSE) {
         i = "You can set {.arg recycle} to {.code TRUE}."
       ))
     }
-    value <- rep_len(values, length.out = i)[[i]]
+    remainder <- i %% length(values)
+    if (remainder == 0) {
+      index <- length(values)
+    } else {
+      index <- remainder
+    }
+    value <- rep_len(values, length.out = index)[[index]]
     i <<- i + 1
     value
   }