tests/testthat/test-utils.R

In kmknn: k-Means for k-Nearest Neighbors

# Tests the utilities.
# library(kmknn); library(testthat); source("test-utils.R")

set.seed(80000)
test_that("worker job assignment works correctly", {
    jobs <- sample(1000)
    by.core <- kmknn:::.assign_jobs(jobs, SerialParam())
    expect_identical(list(jobs), by.core)

    for (x in 2:10) { 
        by.core <- kmknn:::.assign_jobs(jobs, SnowParam(x))
        expect_identical(length(by.core), x)
        expect_identical(jobs, unlist(by.core))
    }

    # Empty inputs.
    jobs <- integer(0)
    by.core <- kmknn:::.assign_jobs(jobs, SerialParam())
    expect_identical(list(jobs), by.core)
    by.core <- kmknn:::.assign_jobs(jobs, SnowParam(2))
    expect_identical(list(integer(0), integer(0)), by.core)
})

set.seed(80001)
test_that("subset index conversion works correctly", {
    A <- matrix(0, 20, 10)
    rownames(A) <- paste0("WHEE", seq_len(nrow(A)))
    colnames(A) <- paste0("YAY", seq_len(ncol(A)))

    # Integer by row.
    i <- sample(nrow(A))
    expect_identical(i, kmknn:::.subset_to_index(i, A, byrow=TRUE))
    expect_error(kmknn:::.subset_to_index(i, A, byrow=FALSE), "out of range")
    expect_identical(integer(0), kmknn:::.subset_to_index(integer(0), A, byrow=TRUE))

    # Integer by column.
    i <- sample(ncol(A), 15, replace=TRUE)
    expect_identical(i, kmknn:::.subset_to_index(i, A, byrow=FALSE))
    expect_identical(integer(0), kmknn:::.subset_to_index(integer(0), A, byrow=FALSE))

    # Logical by row
    j <- rbinom(nrow(A), 1, 0.5)==1
    expect_identical(which(j), kmknn:::.subset_to_index(j, A, byrow=TRUE))
    expect_error(kmknn:::.subset_to_index(j, A, byrow=FALSE), "out of range")
    expect_identical(integer(0), kmknn:::.subset_to_index(logical(0), A, byrow=TRUE))

    # Logical by column.
    j <- rbinom(ncol(A), 1, 0.5)==1
    expect_identical(which(j), kmknn:::.subset_to_index(j, A, byrow=FALSE))
    expect_identical(integer(0), kmknn:::.subset_to_index(logical(0), A, byrow=FALSE))

    # Character by row.
    k <- sample(rownames(A), 32, replace=TRUE)
    expect_identical(match(k, rownames(A)), kmknn:::.subset_to_index(k, A, byrow=TRUE))
    expect_error(kmknn:::.subset_to_index(k, A, byrow=FALSE), "out of range")
    expect_identical(integer(0), kmknn:::.subset_to_index(character(0), A, byrow=TRUE))

    # Character by column.
    k <- sample(colnames(A), 12, replace=TRUE)
    expect_identical(match(k, colnames(A)), kmknn:::.subset_to_index(k, A, byrow=FALSE))
    expect_error(kmknn:::.subset_to_index(k, A, byrow=TRUE), "out of range")
    expect_identical(integer(0), kmknn:::.subset_to_index(character(0), A, byrow=FALSE))
})