R Under development (unstable) (2023-11-06 r85483 ucrt) -- "Unsuffered Consequences" Copyright (C) 2023 The R Foundation for Statistical Computing Platform: x86_64-w64-mingw32/x64 R is free software and comes with ABSOLUTELY NO WARRANTY. You are welcome to redistribute it under certain conditions. Type 'license()' or 'licence()' for distribution details. R is a collaborative project with many contributors. Type 'contributors()' for more information and 'citation()' on how to cite R or R packages in publications. Type 'demo()' for some demos, 'help()' for on-line help, or 'help.start()' for an HTML browser interface to help. Type 'q()' to quit R. > library("matrixStats") > > ## Create isFALSE() if running on an old version of R > if (!exists("isFALSE", mode="function")) { + isFALSE <- function(x) is.logical(x) && length(x) == 1L && !is.na(x) && !x + } > > rowWeightedMeans_R <- function(x, w, na.rm = FALSE, ..., useNames = NA) { + res <- apply(x, MARGIN = 1L, FUN = weighted.mean, w = w, na.rm = na.rm, ...) + + # Keep naming support consistency same as rowWeightedMeans() + idxs <- which(is.na(w) | w != 0) + nw <- length(idxs) + if (na.rm) na.rm <- anyMissing(x) + if ((!is.null(w) && nw == 0L) || isFALSE(na.rm)) { + if (is.na(useNames) || !useNames) names(res) <- NULL + } + else if (isFALSE(useNames)) names(res) <- NULL + + res + } > > colWeightedMeans_R <- function(x, w, na.rm = FALSE, ..., useNames = NA) { + res <- apply(x, MARGIN = 1L, FUN = weighted.mean, w = w, na.rm = na.rm, ...) + + # Keep naming support consistency same as colWeightedMeans() + idxs <- which(is.na(w) | w != 0) + nw <- length(idxs) + if (!is.null(w) && nw == 0L) { + if (is.na(useNames) || !useNames) names(res) <- NULL + } + else if (isFALSE(useNames)) names(res) <- NULL + + res + } > > > # - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - > # Subsetted tests > # - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - > source("utils/validateIndicesFramework.R") > # To check names attribute > dimnames <- list(letters[1:6], LETTERS[1:6]) > for (mode in c("numeric", "integer", "logical")) { + x <- matrix(runif(6 * 6, min = -6, max = 6), nrow = 6, ncol = 6) + w <- runif(6, min = 0, max = 6) + storage.mode(x) <- mode + storage.mode(w) <- if (mode == "logical") "integer" else mode + if (mode == "numeric") w[1] <- Inf + + # Test with and without dimnames on x + for (setDimnames in c(TRUE, FALSE)) { + if (setDimnames) dimnames(x) <- dimnames + else dimnames(x) <- NULL + + count <- 0L + for (rows in index_cases) { + for (cols in index_cases) { + count <- count + 1L + na.rm <- c(TRUE, FALSE)[count %% 2 + 1] + useNames <- c(if (!matrixStats:::isUseNamesNADefunct()) NA, TRUE, FALSE) + useNames <- useNames[count %% length(useNames) + 1] + + validateIndicesTestMatrix_w(x, w, rows, cols, + ftest = rowWeightedMeans, fsure = rowWeightedMeans_R, + na.rm = na.rm, useNames = useNames) + validateIndicesTestMatrix_w(x, w, rows, cols, + fcoltest = colWeightedMeans, fsure = colWeightedMeans_R, + na.rm = na.rm, useNames = useNames) + } + } + } + } > > proc.time() user system elapsed 1.68 0.09 1.76