1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
library("matrixStats")
rowCollapse_R <- function(x, idxs, ..., useNames = TRUE) {
ans <- c()
storage.mode(ans) <- storage.mode(x)
for (ii in seq_len(length(idxs))) {
ans[ii] <- x[ii, idxs[ii]]
}
# Preserve names attribute
if (useNames) {
names <- rownames(x)
if (!is.null(names)) names(ans) <- names
}
ans
}
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
# Subsetted tests
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
source("utils/validateIndicesFramework.R")
x <- matrix(runif(6 * 6, min = -6, max = 6), nrow = 6, ncol = 6)
storage.mode(x) <- "integer"
# To check names attribute
dimnames <- list(letters[1:6], LETTERS[1:6])
for (rows in index_cases) {
if (is.null(rows)) rows <- seq_len(nrow(x))
for (idxs in list(2L, seq_len(6L))) {
for (useNames in c(TRUE, FALSE)) {
suppressWarnings({
actual <- tryCatch(rowCollapse(x, idxs, rows = rows, useNames = useNames),
error = function(c) "error")
expect <- tryCatch({
idxs_0 <- rep(idxs, length.out = nrow(x))[rows]
rowCollapse_R(x[rows, , drop = FALSE], idxs_0, useNames = useNames)
}, error = function(c) "error")
})
stopifnot(all.equal(actual, expect))
suppressWarnings({
actual <- tryCatch(colCollapse(t(x), idxs, cols = rows, useNames = useNames),
error = function(c) "error")
})
stopifnot(all.equal(actual, expect))
# Check names attribute
dimnames(x) <- dimnames
suppressWarnings({
actual <- tryCatch(rowCollapse(x, idxs, rows = rows, useNames = useNames),
error = function(c) "error")
expect <- tryCatch({
idxs_0 <- rep(idxs, length.out = nrow(x))[rows]
rowCollapse_R(x[rows, , drop = FALSE], idxs_0, useNames = useNames)
}, error = function(c) "error")
})
stopifnot(all.equal(actual, expect))
suppressWarnings({
actual <- tryCatch(colCollapse(t(x), idxs, cols = rows, useNames = useNames),
error = function(c) "error")
})
stopifnot(all.equal(actual, expect))
dimnames(x) <- NULL
}
}
}
|
