tests/testthat/test-rbitr_helpers.R

In dryguy/rbitr: Arbiter: One who ensures adherence to the rules and laws of chess

list_1 <- list(
  list(list(1, 2, 3), list(4, 5, 6)),
  list(list(7, 8, 9), list(10, 11, 12))
)
list_2 <- list(
  list(c(1, 2, 3), c(4, 5, 6)),
  list(c(7, 8, 9), c(10, 11, 12))
)
list_3 <- list(
  list(list(1, 2, 3), list(4, 5, 6)),
  list(list(7, 8, 9), numeric(0))
)
list_4 <- list(
  list(list(1, 2, 3), list(4, 5, 6)),
  list(list(7, 8, 9), NA)
)
test_that('nested_lapply applies input_function to root levels of x', {
  result <- nested_lapply(list_1, unlist)
  expect_identical(result, list_2)
  replace_empty_vectors_with_NA <- function(y) {
    if (length(y) == 0) {NA} else {y}
  }
  result <- nested_lapply(list_3, replace_empty_vectors_with_NA)
  expect_identical(result, list_4)
})


dataframe <- data.frame(
  x = 1:4,
  y = c(1, -1, 1, -1)
)
test_that('add_x_intercepts adds x-intercepts', {
  new_dataframe <- add_x_intercepts(dataframe, 'x', 'y')
  expect_equal(new_dataframe$x, c(1, 1.5, 2, 2.5, 3, 3.5, 4))
})


test_that('plot_2_color_area returns a ggplot', {
  p1 <- plot_2_color_area(dataframe, 'x', 'y', background = 'none')
  expect_equal(class(p1), c('gg', 'ggplot'))
  p1 <- plot_2_color_area(dataframe, 'x', 'y', background = 'gradient')
  expect_equal(class(p1), c('gg', 'ggplot'))
})

test_that('inv_scale_move_times calculates move times from scaled move times', {
  expect_equal(inv_winning_chances(winning_chances(10)), 10)
})

test_that('inv_winning_chances calculates scores from winning chances', {
  expect_equal(inv_scale_move_times(scale_move_times(10)), 10)
})