Nothing
tests/testthat/test-partition.R
In nuggets: Extensible Data Pattern Searching Framework
test_that("partition basics", {
expect_equal(partition(data.frame()),
tibble())
expect_equal(partition(data.frame(a = c(T, T, F, F)), where(is.raw)),
tibble(a = c(T, T, F, F)))
expect_equal(partition(data.frame(a = c(T, T, F, F)),
.keep = FALSE),
tibble("a=T" = c(T, T, F, F),
"a=F" = c(F, F, T, T)))
expect_equal(partition(data.frame(a = c(T, T, F, NA)),
.na = TRUE,
.keep = FALSE),
tibble("a=T" = c(T, T, F, F),
"a=F" = c(F, F, T, F),
"a=NA" = c(F, F, F, T)))
expect_equal(partition(data.frame(a = c(T, T, F, NA)),
.na = FALSE,
.keep = FALSE),
tibble("a=T" = c(T, T, F, F),
"a=F" = c(F, F, T, F)))
expect_equal(partition(data.frame(a = c(T, T, F, F)),
.keep = TRUE),
tibble("a" = c(T, T, F, F),
"a=T" = c(T, T, F, F),
"a=F" = c(F, F, T, T)))
expect_equal(partition(data.frame(a = factor(c("a", "b", "b", "c"))),
.keep = FALSE),
tibble("a=a" = c(T, F, F, F),
"a=b" = c(F, T, T, F),
"a=c" = c(F, F, F, T)))
expect_equal(partition(data.frame(a = factor(c("a", "b", "b", "c"))),
.keep = TRUE),
tibble("a" = factor(c("a", "b", "b", "c")),
"a=a" = c(T, F, F, F),
"a=b" = c(F, T, T, F),
"a=c" = c(F, F, F, T)))
expect_equal(partition(data.frame(a = c(T, T, F, F),
b = c(F, F, F, F),
c = c(T, T, T, T),
d = factor(c("a", "b", "b", "c"))),
b:c,
.keep = FALSE),
tibble(a = c(T, T, F, F),
d = factor(c("a", "b", "b", "c")),
"b=T" = c(F, F, F, F),
"b=F" = c(T, T, T, T),
"c=T" = c(T, T, T, T),
"c=F" = c(F, F, F, F)))
expect_equal(partition(data.frame(a = c(T, T, F, F),
b = c(F, F, F, F),
c = c(T, T, T, T),
d = factor(c("a", "b", "b", "c"))),
b, c,
.keep = FALSE),
tibble(a = c(T, T, F, F),
d = factor(c("a", "b", "b", "c")),
"b=T" = c(F, F, F, F),
"b=F" = c(T, T, T, T),
"c=T" = c(T, T, T, T),
"c=F" = c(F, F, F, F)))
expect_equal(partition(data.frame(a = c(T, T, F, F),
b = c(F, F, F, F),
c = c(T, T, T, T),
d = factor(c("a", "b", "b", "c"))),
b, c,
.keep = TRUE),
tibble(a = c(T, T, F, F),
b = c(F, F, F, F),
c = c(T, T, T, T),
d = factor(c("a", "b", "b", "c")),
"b=T" = c(F, F, F, F),
"b=F" = c(T, T, T, T),
"c=T" = c(T, T, T, T),
"c=F" = c(F, F, F, F)))
})
test_that("partition crisp", {
expect_equal(partition(data.frame(a = 0:10),
.breaks = 2,
.keep = FALSE,
.method = "crisp"),
tibble("a=(-Inf;5]" = c(T,T,T,T,T,T,F,F,F,F,F),
"a=(5;Inf]" = c(F,F,F,F,F,F,T,T,T,T,T)))
expect_equal(partition(data.frame(b = 1:10),
.breaks = 3,
.keep = FALSE,
.method = "crisp"),
tibble("b=(-Inf;4]" = c(T,T,T,T,F,F,F,F,F,F),
"b=(4;7]" = c(F,F,F,F,T,T,T,F,F,F),
"b=(7;Inf]" = c(F,F,F,F,F,F,F,T,T,T)))
expect_equal(partition(data.frame(b = 1:10),
.breaks = c(-Inf, 4, 7, Inf),
.keep = FALSE,
.method = "crisp",
.right = TRUE),
tibble("b=(-Inf;4]" = c(T,T,T,T,F,F,F,F,F,F),
"b=(4;7]" = c(F,F,F,F,T,T,T,F,F,F),
"b=(7;Inf]" = c(F,F,F,F,F,F,F,T,T,T)))
expect_equal(partition(data.frame(b = 1:10),
.breaks = c(4, 7),
.keep = FALSE,
.na = TRUE,
.method = "crisp",
.right = FALSE),
tibble("b=[4;7)" = c(F,F,F,T,T,T,F,F,F,F)))
expect_equal(partition(data.frame(b = c(1:10, NA, NA)),
.breaks = c(4, 7),
.keep = FALSE,
.na = TRUE,
.method = "crisp",
.right = FALSE),
tibble("b=[4;7)" = c(F,F,F,T,T,T,F,F,F,F,F,F),
"b=NA" = c(F,F,F,F,F,F,F,F,F,F,T,T)))
expect_equal(partition(data.frame(b = c(1:10, NA, NA)),
.breaks = c(4, 7),
.keep = FALSE,
.na = FALSE,
.method = "crisp",
.right = FALSE),
tibble("b=[4;7)" = c(F,F,F,T,T,T,F,F,F,F,F,F)))
expect_equal(partition(data.frame(b = 1:10),
.breaks = c(-Inf, 4, 7, Inf),
.keep = FALSE,
.method = "crisp",
.labels = c("A", "BBB", "cc"),
.right = TRUE),
tibble("b=A" = c(T,T,T,T,F,F,F,F,F,F),
"b=BBB" = c(F,F,F,F,T,T,T,F,F,F),
"b=cc" = c(F,F,F,F,F,F,F,T,T,T)))
expect_equal(partition(data.frame(b = 1:10),
.breaks = 3,
.keep = FALSE,
.method = "crisp",
.labels = c("A", "BBB", "cc"),
.right = TRUE),
tibble("b=A" = c(T,T,T,T,F,F,F,F,F,F),
"b=BBB" = c(F,F,F,F,T,T,T,F,F,F),
"b=cc" = c(F,F,F,F,F,F,F,T,T,T)))
expect_error(partition(data.frame(a = 0:10),
.method = "crisp"),
"`.breaks` must not be NULL in order to partition numeric column `a`")
expect_error(partition(data.frame(a = 0:10),
.breaks = -1,
.method = "crisp"),
"If `.breaks` is a single value, it must be a natural number greater than 1.")
expect_error(partition(data.frame(a = 0:10),
.breaks = 1.5,
.method = "crisp"),
"If `.breaks` is a single value, it must be a natural number greater than 1.")
expect_error(partition(data.frame(b = 1:10),
.breaks = 2,
.method = "crisp",
.labels = c("A", "BBB", "cc")),
"If `.breaks` is scalar, the length of `.labels` must be equal to the value of `.breaks`.")
expect_error(partition(data.frame(b = 1:10),
.breaks = c(-Inf, 4, 7, Inf),
.method = "crisp",
.labels = c("A", "cc")),
"If `.breaks` is non-scalar, the length of `.labels` must be equal to the length of `.breaks` - 1.")
})
test_that("partition triangle", {
expect_equal(partition(data.frame(a = 0:10),
.breaks = 2,
.keep = FALSE,
.method = "triangle"),
tibble("a=(-Inf;0;10)" = seq(1, 0, length.out = 11),
"a=(0;10;Inf)" = seq(0, 1, length.out = 11)))
expect_equal(partition(data.frame(a = 0:10),
.breaks = 3,
.keep = FALSE,
.method = "triangle"),
tibble("a=(-Inf;0;5)" = c(1.0, 0.8, 0.6, 0.4, 0.2, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0),
"a=(0;5;10)" = c(0.0, 0.2, 0.4, 0.6, 0.8, 1.0, 0.8, 0.6, 0.4, 0.2, 0.0),
"a=(5;10;Inf)" = c(0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.2, 0.4, 0.6, 0.8, 1.0)))
expect_error(partition(data.frame(a = 0:10),
.breaks = 1,
.keep = FALSE,
.method = "triangle"),
"If `.breaks` is a single value, it must be a natural number greater than 1.")
expect_error(partition(data.frame(a = 0:10),
.breaks = c(2, 5),
.keep = FALSE,
.method = "triangle"),
"If `.breaks` is non-scalar, it must be a vector with at least 3 elements.")
expect_error(partition(data.frame(b = 1:10),
.breaks = 2,
.method = "triangle",
.labels = c("A", "BBB", "cc")),
"If `.breaks` is scalar, the length of `.labels` must be equal to the value of `.breaks`.")
expect_error(partition(data.frame(b = 1:10),
.breaks = c(-Inf, 4, 7, Inf),
.method = "triangle",
.labels = c("A", "b", "cc")),
"If `.breaks` is non-scalar, the length of `.labels` must be equal to the length of `.breaks` - 2.")
})
test_that("partition raisedcos", {
res <- partition(data.frame(a = 0:10),
.breaks = 2,
.keep = FALSE,
.method = "raisedcos")
expect_equal(names(res), c("a=(-Inf;0;10)", "a=(0;10;Inf)"))
expect_equal(res[[1]],
c(1.00000000, 0.97552826, 0.90450850, 0.79389263, 0.65450850,
0.50000000, 0.34549150, 0.20610737, 0.09549150, 0.02447174,
0.00000000))
expect_equal(res[[2]],
c(0.00000000, 0.02447174, 0.09549150, 0.20610737, 0.34549150,
0.50000000, 0.65450850, 0.79389263, 0.90450850, 0.97552826,
1.00000000))
expect_error(partition(data.frame(a = 0:10),
.breaks = 1,
.keep = FALSE,
.method = "raisedcos"),
"If `.breaks` is a single value, it must be a natural number greater than 1.")
expect_error(partition(data.frame(a = 0:10),
.breaks = c(2, 5),
.keep = FALSE,
.method = "raisedcos"),
"If `.breaks` is non-scalar, it must be a vector with at least 3 elements.")
expect_error(partition(data.frame(b = 1:10),
.breaks = 2,
.method = "raisedcos",
.labels = c("A", "BBB", "cc")),
"If `.breaks` is scalar, the length of `.labels` must be equal to the value of `.breaks`.")
expect_error(partition(data.frame(b = 1:10),
.breaks = c(-Inf, 4, 7, Inf),
.method = "raisedcos",
.labels = c("A", "b", "cc")),
"If `.breaks` is non-scalar, the length of `.labels` must be equal to the length of `.breaks` - 2.")
})
test_that("partition errors", {
d <- data.frame(a = factor(c("a", "a", "b", "b", "a")))
expect_error(partition(as.list(d), a),
"`.data` must be a data frame")
expect_error(partition(d, x),
"Can't select columns that don't exist")
expect_error(partition(data.frame(a = 0:10),
.breaks = "x",
.method = "crisp"),
"`.breaks` must be a numeric vector or NULL")
expect_error(partition(d, a, .labels = 1:5),
"`.labels` must be a character vector")
expect_error(partition(d, a, .na = "TRUE"),
"`.na` must be a flag")
expect_error(partition(d, a, .keep = "TRUE"),
"`.keep` must be a flag")
expect_error(partition(d, a, .right = "TRUE"),
"`.right` must be a flag")
expect_error(partition(d, a, .method = "foo"),
'`.method` must be equal to one of: "crisp", "triangle", "raisedcos".')
})
Try the nuggets package in your browser
Any scripts or data that you put into this service are public.
nuggets documentation built on April 3, 2025, 8:07 p.m.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
test_that("partition basics", {
expect_equal(partition(data.frame()),
tibble())
expect_equal(partition(data.frame(a = c(T, T, F, F)), where(is.raw)),
tibble(a = c(T, T, F, F)))
expect_equal(partition(data.frame(a = c(T, T, F, F)),
.keep = FALSE),
tibble("a=T" = c(T, T, F, F),
"a=F" = c(F, F, T, T)))
expect_equal(partition(data.frame(a = c(T, T, F, NA)),
.na = TRUE,
.keep = FALSE),
tibble("a=T" = c(T, T, F, F),
"a=F" = c(F, F, T, F),
"a=NA" = c(F, F, F, T)))
expect_equal(partition(data.frame(a = c(T, T, F, NA)),
.na = FALSE,
.keep = FALSE),
tibble("a=T" = c(T, T, F, F),
"a=F" = c(F, F, T, F)))
expect_equal(partition(data.frame(a = c(T, T, F, F)),
.keep = TRUE),
tibble("a" = c(T, T, F, F),
"a=T" = c(T, T, F, F),
"a=F" = c(F, F, T, T)))
expect_equal(partition(data.frame(a = factor(c("a", "b", "b", "c"))),
.keep = FALSE),
tibble("a=a" = c(T, F, F, F),
"a=b" = c(F, T, T, F),
"a=c" = c(F, F, F, T)))
expect_equal(partition(data.frame(a = factor(c("a", "b", "b", "c"))),
.keep = TRUE),
tibble("a" = factor(c("a", "b", "b", "c")),
"a=a" = c(T, F, F, F),
"a=b" = c(F, T, T, F),
"a=c" = c(F, F, F, T)))
expect_equal(partition(data.frame(a = c(T, T, F, F),
b = c(F, F, F, F),
c = c(T, T, T, T),
d = factor(c("a", "b", "b", "c"))),
b:c,
.keep = FALSE),
tibble(a = c(T, T, F, F),
d = factor(c("a", "b", "b", "c")),
"b=T" = c(F, F, F, F),
"b=F" = c(T, T, T, T),
"c=T" = c(T, T, T, T),
"c=F" = c(F, F, F, F)))
expect_equal(partition(data.frame(a = c(T, T, F, F),
b = c(F, F, F, F),
c = c(T, T, T, T),
d = factor(c("a", "b", "b", "c"))),
b, c,
.keep = FALSE),
tibble(a = c(T, T, F, F),
d = factor(c("a", "b", "b", "c")),
"b=T" = c(F, F, F, F),
"b=F" = c(T, T, T, T),
"c=T" = c(T, T, T, T),
"c=F" = c(F, F, F, F)))
expect_equal(partition(data.frame(a = c(T, T, F, F),
b = c(F, F, F, F),
c = c(T, T, T, T),
d = factor(c("a", "b", "b", "c"))),
b, c,
.keep = TRUE),
tibble(a = c(T, T, F, F),
b = c(F, F, F, F),
c = c(T, T, T, T),
d = factor(c("a", "b", "b", "c")),
"b=T" = c(F, F, F, F),
"b=F" = c(T, T, T, T),
"c=T" = c(T, T, T, T),
"c=F" = c(F, F, F, F)))
})
test_that("partition crisp", {
expect_equal(partition(data.frame(a = 0:10),
.breaks = 2,
.keep = FALSE,
.method = "crisp"),
tibble("a=(-Inf;5]" = c(T,T,T,T,T,T,F,F,F,F,F),
"a=(5;Inf]" = c(F,F,F,F,F,F,T,T,T,T,T)))
expect_equal(partition(data.frame(b = 1:10),
.breaks = 3,
.keep = FALSE,
.method = "crisp"),
tibble("b=(-Inf;4]" = c(T,T,T,T,F,F,F,F,F,F),
"b=(4;7]" = c(F,F,F,F,T,T,T,F,F,F),
"b=(7;Inf]" = c(F,F,F,F,F,F,F,T,T,T)))
expect_equal(partition(data.frame(b = 1:10),
.breaks = c(-Inf, 4, 7, Inf),
.keep = FALSE,
.method = "crisp",
.right = TRUE),
tibble("b=(-Inf;4]" = c(T,T,T,T,F,F,F,F,F,F),
"b=(4;7]" = c(F,F,F,F,T,T,T,F,F,F),
"b=(7;Inf]" = c(F,F,F,F,F,F,F,T,T,T)))
expect_equal(partition(data.frame(b = 1:10),
.breaks = c(4, 7),
.keep = FALSE,
.na = TRUE,
.method = "crisp",
.right = FALSE),
tibble("b=[4;7)" = c(F,F,F,T,T,T,F,F,F,F)))
expect_equal(partition(data.frame(b = c(1:10, NA, NA)),
.breaks = c(4, 7),
.keep = FALSE,
.na = TRUE,
.method = "crisp",
.right = FALSE),
tibble("b=[4;7)" = c(F,F,F,T,T,T,F,F,F,F,F,F),
"b=NA" = c(F,F,F,F,F,F,F,F,F,F,T,T)))
expect_equal(partition(data.frame(b = c(1:10, NA, NA)),
.breaks = c(4, 7),
.keep = FALSE,
.na = FALSE,
.method = "crisp",
.right = FALSE),
tibble("b=[4;7)" = c(F,F,F,T,T,T,F,F,F,F,F,F)))
expect_equal(partition(data.frame(b = 1:10),
.breaks = c(-Inf, 4, 7, Inf),
.keep = FALSE,
.method = "crisp",
.labels = c("A", "BBB", "cc"),
.right = TRUE),
tibble("b=A" = c(T,T,T,T,F,F,F,F,F,F),
"b=BBB" = c(F,F,F,F,T,T,T,F,F,F),
"b=cc" = c(F,F,F,F,F,F,F,T,T,T)))
expect_equal(partition(data.frame(b = 1:10),
.breaks = 3,
.keep = FALSE,
.method = "crisp",
.labels = c("A", "BBB", "cc"),
.right = TRUE),
tibble("b=A" = c(T,T,T,T,F,F,F,F,F,F),
"b=BBB" = c(F,F,F,F,T,T,T,F,F,F),
"b=cc" = c(F,F,F,F,F,F,F,T,T,T)))
expect_error(partition(data.frame(a = 0:10),
.method = "crisp"),
"`.breaks` must not be NULL in order to partition numeric column `a`")
expect_error(partition(data.frame(a = 0:10),
.breaks = -1,
.method = "crisp"),
"If `.breaks` is a single value, it must be a natural number greater than 1.")
expect_error(partition(data.frame(a = 0:10),
.breaks = 1.5,
.method = "crisp"),
"If `.breaks` is a single value, it must be a natural number greater than 1.")
expect_error(partition(data.frame(b = 1:10),
.breaks = 2,
.method = "crisp",
.labels = c("A", "BBB", "cc")),
"If `.breaks` is scalar, the length of `.labels` must be equal to the value of `.breaks`.")
expect_error(partition(data.frame(b = 1:10),
.breaks = c(-Inf, 4, 7, Inf),
.method = "crisp",
.labels = c("A", "cc")),
"If `.breaks` is non-scalar, the length of `.labels` must be equal to the length of `.breaks` - 1.")
})
test_that("partition triangle", {
expect_equal(partition(data.frame(a = 0:10),
.breaks = 2,
.keep = FALSE,
.method = "triangle"),
tibble("a=(-Inf;0;10)" = seq(1, 0, length.out = 11),
"a=(0;10;Inf)" = seq(0, 1, length.out = 11)))
expect_equal(partition(data.frame(a = 0:10),
.breaks = 3,
.keep = FALSE,
.method = "triangle"),
tibble("a=(-Inf;0;5)" = c(1.0, 0.8, 0.6, 0.4, 0.2, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0),
"a=(0;5;10)" = c(0.0, 0.2, 0.4, 0.6, 0.8, 1.0, 0.8, 0.6, 0.4, 0.2, 0.0),
"a=(5;10;Inf)" = c(0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.2, 0.4, 0.6, 0.8, 1.0)))
expect_error(partition(data.frame(a = 0:10),
.breaks = 1,
.keep = FALSE,
.method = "triangle"),
"If `.breaks` is a single value, it must be a natural number greater than 1.")
expect_error(partition(data.frame(a = 0:10),
.breaks = c(2, 5),
.keep = FALSE,
.method = "triangle"),
"If `.breaks` is non-scalar, it must be a vector with at least 3 elements.")
expect_error(partition(data.frame(b = 1:10),
.breaks = 2,
.method = "triangle",
.labels = c("A", "BBB", "cc")),
"If `.breaks` is scalar, the length of `.labels` must be equal to the value of `.breaks`.")
expect_error(partition(data.frame(b = 1:10),
.breaks = c(-Inf, 4, 7, Inf),
.method = "triangle",
.labels = c("A", "b", "cc")),
"If `.breaks` is non-scalar, the length of `.labels` must be equal to the length of `.breaks` - 2.")
})
test_that("partition raisedcos", {
res <- partition(data.frame(a = 0:10),
.breaks = 2,
.keep = FALSE,
.method = "raisedcos")
expect_equal(names(res), c("a=(-Inf;0;10)", "a=(0;10;Inf)"))
expect_equal(res[[1]],
c(1.00000000, 0.97552826, 0.90450850, 0.79389263, 0.65450850,
0.50000000, 0.34549150, 0.20610737, 0.09549150, 0.02447174,
0.00000000))
expect_equal(res[[2]],
c(0.00000000, 0.02447174, 0.09549150, 0.20610737, 0.34549150,
0.50000000, 0.65450850, 0.79389263, 0.90450850, 0.97552826,
1.00000000))
expect_error(partition(data.frame(a = 0:10),
.breaks = 1,
.keep = FALSE,
.method = "raisedcos"),
"If `.breaks` is a single value, it must be a natural number greater than 1.")
expect_error(partition(data.frame(a = 0:10),
.breaks = c(2, 5),
.keep = FALSE,
.method = "raisedcos"),
"If `.breaks` is non-scalar, it must be a vector with at least 3 elements.")
expect_error(partition(data.frame(b = 1:10),
.breaks = 2,
.method = "raisedcos",
.labels = c("A", "BBB", "cc")),
"If `.breaks` is scalar, the length of `.labels` must be equal to the value of `.breaks`.")
expect_error(partition(data.frame(b = 1:10),
.breaks = c(-Inf, 4, 7, Inf),
.method = "raisedcos",
.labels = c("A", "b", "cc")),
"If `.breaks` is non-scalar, the length of `.labels` must be equal to the length of `.breaks` - 2.")
})
test_that("partition errors", {
d <- data.frame(a = factor(c("a", "a", "b", "b", "a")))
expect_error(partition(as.list(d), a),
"`.data` must be a data frame")
expect_error(partition(d, x),
"Can't select columns that don't exist")
expect_error(partition(data.frame(a = 0:10),
.breaks = "x",
.method = "crisp"),
"`.breaks` must be a numeric vector or NULL")
expect_error(partition(d, a, .labels = 1:5),
"`.labels` must be a character vector")
expect_error(partition(d, a, .na = "TRUE"),
"`.na` must be a flag")
expect_error(partition(d, a, .keep = "TRUE"),
"`.keep` must be a flag")
expect_error(partition(d, a, .right = "TRUE"),
"`.right` must be a flag")
expect_error(partition(d, a, .method = "foo"),
'`.method` must be equal to one of: "crisp", "triangle", "raisedcos".')
})
Try the nuggets package in your browser
Any scripts or data that you put into this service are public.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Embedding an R snippet on your website
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.