tests/testthat/test_preprocess.R

In citrus: Customer Intelligence Tool for Rapid Understandable Segmentation

library(citrus)
library(testthat)
library(dplyr)

transactional_data <- citrus::transactional_data
data <- transactional_data %>% select(c('transactionid', 'transactionvalue', 'id', 'orderdate'))
output_preprocess <- citrus::preprocess(data, numeric_operation_list = 'mean')
output_with_response <- citrus::preprocess(data, numeric_operation_list = 'mean', target = 'transactionvalue')
output_preprocess_na <- citrus::preprocess(data, numeric_operation_list = NULL)

test_that("Number of Columns", {
  expect_equal(ncol(output_preprocess), 5)
})

test_that("String ID Check", {
  expect_true(is.character(typeof(output_preprocess$id)))
})

test_that("Passing NA to numeric_operations_list defaults to RFM", {
  expect_equal(ncol(output_preprocess_na), 4)
  expect_equal(sort(colnames(output_preprocess_na)), sort(c('id', 'recency', 'frequency', 'monetary')))
})

test_that("Correct Labelling", {
  expect_equal(colnames(output_preprocess), c('id', 'recency', 'frequency', 'monetary', 'transactionvalue_mean'))
})

test_that("Custom target included", {
  expect_true('response' %in% colnames(output_with_response))
})