tests/testthat/test_bin_class.R

In Zelazny7/onx:

context("  Bin Class")


test_that("Bin constructors work", {

  expect_s4_class(BinNumeric(lower=0, upper=100), "BinNumeric")
  expect_error(BinNumeric(lower=10, upper=10))
  expect_error(BinNumeric(lower=10, upper=0))

  expect_equal(BinNumeric()@lower, -Inf)
  expect_equal(BinNumeric()@upper, Inf)

  expect_s4_class(BinFactor(level="level1"), "BinFactor")
  expect_equal(BinFactor()@level, "")

  expect_s4_class(BinMissing(), "BinMissing")

})

test_that("Boolean masks", {

  a <- BinNumeric(lower=-5, upper=5)
  expect_equal(get_boolean_mask(a, 1:10), c(T, T, T, T, T, F, F, F, F, F))
  expect_equal(get_boolean_mask(a, -5:0), c(F, T, T, T, T, T))

  b <- BinFactor(level="cat")
  expect_equal(get_boolean_mask(b, factor(c("dog", "cat"))), c(F, T))
  expect_equal(get_boolean_mask(b, factor(c("cat", "cat"))), c(T, T))
  expect_equal(get_boolean_mask(b, factor(c("CAT", "CAT"))), c(F, F))

  c <- BinMissing()
  expect_equal(get_boolean_mask(c, c(1, 2, NA)), c(F, F, T))
  expect_equal(get_boolean_mask(c, c(NA, NA, NA)), c(T, T, T))
  expect_equal(get_boolean_mask(c, factor(c(NA, "cat"))), c(T, F))


})

test_that("Combine", {

  a <- BinNumeric(lower=1, upper=10)
  b <- BinNumeric(lower=7, upper=15)
  c <- BinNumeric(lower=15, upper=20)

  ##--- Combining disparate and overlapping numeric bins
  expect_equal(combinable(a, b), TRUE)
  expect_equal(combinable(b, a), TRUE)
  expect_equal(combinable(a, c), FALSE)
  expect_equal(combinable(c, a), FALSE)

  expect_equal(combinable(a, BinMissing()), FALSE)
  expect_equal(combinable(a, BinException(exception=-1)), FALSE)


  expect_equal(combine(a, b), list(BinNumeric(lower=1, upper=15)))
  expect_equal(combine(a, c), list(a, c))

  e <- BinException(exception=-1)
  f <- BinFactor(level="z")
  m <- BinMissing()

  ##-- Combining diferent bin types should return a list
  expect_equal(combine(a, m), list(a, m))
  expect_equal(combine(a, f), list(a, f))
  expect_equal(combine(a, e), list(a, e))
  expect_equal(combine(f, m), list(f, m))
  expect_equal(combine(f, e), list(f, e))

  A <- BinNumeric(lower=0, upper=20)
  B <- BinNumeric(lower=10, upper=30)
  C <- BinNumeric(lower=25, upper=50)

  expect_equal(combinable(A, B), TRUE)
  expect_equal(combine(A, B), list(BinNumeric(lower=0, upper=30)))

  expect_equal(combinable(A, C), FALSE)
  expect_equal(combine(A, C), list(A, C))

  expect_equal(Reduce(combine, list(A, B, C)), list(BinNumeric(lower=0, upper=50)))
  expect_equal(Reduce(combine, list(A, C, B)), list(BinNumeric(lower=0, upper=50)))

})

test_that("Expand bins",  {

  expect_equal(
    expand(BinNumeric(lower=1, upper=15), 10),
    list(BinNumeric(lower=1, upper=10), BinNumeric(lower=10, upper=15)))

  expect_equal(
    expand(BinNumeric(lower=1, upper=15), c(20)),
    list(BinNumeric(lower=1, upper=15)))

  expect_equal(
    expand(BinNumeric(lower=1, upper=15), c(-20)),
    list(BinNumeric(lower=1, upper=15)))

  expect_equal(
    expand(BinNumeric(lower=1, upper=20), seq(1, 20, 4)),
    list(
      BinNumeric(lower=1, upper=5),
      BinNumeric(lower=5, upper=9),
      BinNumeric(lower=9, upper=13),
      BinNumeric(lower=13, upper=17),
      BinNumeric(lower=17, upper=20)
    ))

  expect_equal(
    Reduce(combine, expand(BinNumeric(lower=1, upper=20), seq(1, 20, 4))),
    list(BinNumeric(lower=1, upper=20)))

})


test_that("Bin labels", {

  a <- BinNumeric(lower=1, upper=10)
  b <- BinFactor(level="Factor Level")
  c <- BinMissing()

  expect_equal(get_label(a), "(1 - 10]")
  expect_equal(get_label(b), "Factor Level")
  expect_equal(get_label(c), "Missing")


})

test_that("Order Values", {

  a <- BinNumeric(lower=1, upper=10)
  b <- BinFactor(level="Factor Level")
  c <- BinMissing()
  d <- BinException(exception=-1)

  expect_equal(ordervalue(a), c(1, 1))
  expect_equal(ordervalue(b), c(1, as.numeric(charToRaw("F"))))
  expect_equal(ordervalue(c), c(3, Inf))
  expect_equal(ordervalue(d), c(2, -1))


})