tests/testthat/test_pareto.R

In CVUA-RRW/NMRrecipes: Recipes Steps for NMR Scaling

library(testthat)
library(rlang)
library(NMRrecipes)

context("Testing pareto scaling")

library(tidyverse)
library(recipes)

iris_num <- iris %>% 
        select(-Species) %>% 
        tibble::tibble()
means <- vapply(iris_num, mean, c(mean=0))
sds <- vapply(iris_num, sd, c(sd=0))
sdroots <- sqrt(sds)

test_that('correct processing', {
    processed <- sweep(as.matrix(iris_num), 2, means, "-") %>%
                 sweep(., 2, sdroots, "/") %>% 
                 tibble::as_tibble()
    
    pareto_res <- recipe(Species ~ ., data = iris) %>%
                  step_pareto(all_predictors()) %>%
                  prep() %>%
                  juice() %>% select(-Species)
    
    expect_equal(processed, pareto_res)
})