File: test_base_summarizeLevels.R

package info (click to toggle)
r-cran-mlr 2.19.1%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 8,392 kB
  • sloc: ansic: 65; sh: 13; makefile: 5
file content (15 lines) | stat: -rwxr-xr-x 747 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

test_that("summarizeLevels", {
  data = data.frame(num = as.numeric(1:6), fac = as.factor(1:6), int = as.integer(1:6))
  data2 = data.frame(cha = as.character(1:6), cha2 = 3:8)
  data3 = data.frame(fac = as.factor(c(rep(1, 1000), rep(2, 100000))), fac2 = as.integer(rep(4, 101000)))
  data4 = data.frame(fac1 = as.factor(1))

  expect_error(summarizeLevels(data, "num"))
  expect_equal(length(summarizeLevels(data)), 1L)
  expect_equal(length(summarizeLevels(data, "fac")), 1L)
  expect_equal(as.numeric(summarizeLevels(data)[[1]]), rep(1, 6))
  expect_equal(as.numeric(summarizeLevels(data2)[[1]]), rep(1, 6))
  expect_equal(as.numeric(summarizeLevels(data3)[[1]]), c(1000, 100000))
  expect_equal(as.numeric(summarizeLevels(data4)[[1]]), 1)
})