1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
test_that("simple numerics", {
set.seed(13333)
x1 <- rnorm(1000)
str1a <- make_strata(x1)
tab1a <- table(str1a)
expect_equal(as.vector(tab1a), rep(250, 4))
expect_warning(str1b <- make_strata(x1, depth = 500), "2 breaks instead")
tab1b <- table(str1b)
expect_equal(as.vector(tab1b), rep(500, 2))
str1c <- make_strata(c(NA, x1[1:999]))
tab1c <- table(str1c)
expect_true(all(as.vector(tab1c) %in% 249:251))
})
test_that("simple character", {
x2 <- factor(rep(LETTERS[1:12], each = 20))
expect_warning(
str2a <- make_strata(x2, pool = 0.05),
"Stratifying groups that make up 5%"
)
expect_equal(table(str2a, dnn = ""), table(x2, dnn = ""))
x2[5] <- NA
expect_warning(
str2b <- make_strata(x2, pool = 0.05),
"Stratifying groups that make up 5%"
)
expect_true(all(as.vector(table(str2b, dnn = "")) %in% 19:21))
})
test_that("bad data", {
x3 <- factor(rep(LETTERS[1:15], each = 50))
expect_warning(make_strata(x3), "Too little data")
expect_snapshot(s1 <- make_strata(x3, pool = 0.06))
expect_snapshot(s2 <- make_strata(mtcars$mpg))
expect_snapshot(s3 <- make_strata(seq_len(50), breaks = -1))
})
# strata_check() ----------------------------------------------------------
test_that("don't stratify on Surv objects", {
df <- data.frame(
time = c(85, 79, 70, 6, 32, 8, 17, 93, 81, 76),
event = c(0, 0, 1, 0, 0, 0, 1, 1, 1, 1)
)
df$surv <- structure(
c(
85, 79, 70, 6, 32, 8, 17, 93, 81, 76,
0, 0, 1, 0, 0, 0, 1, 1, 1, 1
),
.Dim = c(10L, 2L),
.Dimnames = list(NULL, c("time", "status")),
type = "right",
class = "Surv"
)
expect_error(strata_check("surv", df))
})
|
