tests/testthat/test-utils.R

In dispositionEffect: Analysis of Disposition Effect on Financial Portfolios

context("utilities")

test_that("negative match works", {
	expect_false(1 %!in% 1:3)
	expect_true(1 %!in% 4:6)
})

test_that("ewise_mean works", {
	expect_equal(ewise_mean(1:10, 99:90), rep(50, 10))
	expect_equal(ewise_mean(c(NA, 2:10), 99:90, na.rm = TRUE), c(99, rep(50, 9)))
	expect_equal(ewise_mean(c(0, 2:10), 99:90, zero.substitute = TRUE), c(99, rep(50, 9)))
})

df <- data.frame(
	"RG_value" = c(-1, 1, 1, 1, -1, 1),
	"RL_value" = c(-1, 1, -1, -1, 1, -1),
	"PG_value" = c(1, 1, -1, 1, -1, 1),
	"PL_value" = c(-1, -1, -1, 1, 1, -1)
)

test_that("check_gainloss works", {
	expect_type(check_gainloss(df[1,]), "character")
	expect_type(check_gainloss(df[2,]), "character")
	expect_type(check_gainloss(df[3,]), "character")
	expect_type(check_gainloss(df[4,]), "character")
	expect_type(check_gainloss(df[5,]), "character")
	expect_type(check_gainloss(df[6,]), "NULL")
})

test_that("check_values works", {
	expect_equal(check_values(NULL, c("A", "B", "C")), c("A", "B", "C"))
	expect_equal(check_values("A", c("A", "B", "C")), c("B", "C"))
	expect_equal(check_values(c("A", "B"), c("A", "B", "C")), "C")
	expect_equal(check_values(c("A", "B", "C"), c("A", "B", "C")), NULL)

	expect_equal(check_values(c("B", "S"), c("B", "S"), no_exception = TRUE), list("target" = NULL, "input" = NULL))
	expect_equal(check_values(c("B", "S", "Buy"), c("B", "S"), no_exception = TRUE), list("target" = NULL, "input" = "Buy"))
	expect_equal(check_values("B", c("B", "S"), no_exception = TRUE), list("target" = "S", "input" = NULL))
	expect_equal(check_values(c("B", "Buy"), c("B", "S"), no_exception = TRUE), list("target" = "S", "input" = "Buy"))

	expect_equal(check_values("B", c("B", "S"), weak_target = TRUE), NULL)
	expect_equal(check_values("B", c("B", "S"), no_exception = TRUE, weak_target = TRUE), list("target" = NULL, "input" = NULL))
	expect_equal(check_values(c("B", "S", "Buy"), c("B", "S"), no_exception = TRUE, weak_target = TRUE), list("target" = NULL, "input" = "Buy"))
	expect_equal(check_values(c("B", "Buy"), c("B", "S"), no_exception = TRUE, weak_target = TRUE), list("target" = NULL, "input" = "Buy"))
})