AMR/tests/testthat/test-portion.R

168 lines
7.5 KiB
R
Raw Normal View History

2018-08-10 15:01:05 +02:00
context("portion.R")
2018-07-13 17:23:46 +02:00
2018-08-12 17:44:06 +02:00
test_that("portions works", {
2018-08-03 14:49:29 +02:00
# amox resistance in `septic_patients`
2018-08-10 15:01:05 +02:00
expect_equal(portion_R(septic_patients$amox), 0.6603, tolerance = 0.0001)
expect_equal(portion_I(septic_patients$amox), 0.0030, tolerance = 0.0001)
expect_equal(1 - portion_R(septic_patients$amox) - portion_I(septic_patients$amox),
portion_S(septic_patients$amox))
expect_equal(portion_R(septic_patients$amox) + portion_I(septic_patients$amox),
portion_IR(septic_patients$amox))
expect_equal(portion_S(septic_patients$amox) + portion_I(septic_patients$amox),
portion_SI(septic_patients$amox))
2018-03-27 17:43:42 +02:00
2018-04-03 11:08:31 +02:00
# pita+genta susceptibility around 98.09%
2018-07-13 17:23:46 +02:00
expect_equal(suppressWarnings(rsi(septic_patients$pita,
septic_patients$gent,
interpretation = "S")),
0.9535,
2018-04-03 11:08:31 +02:00
tolerance = 0.0001)
2018-05-02 14:56:25 +02:00
2018-08-03 14:49:29 +02:00
# percentages
expect_equal(septic_patients %>%
group_by(hospital_id) %>%
2018-08-10 15:01:05 +02:00
summarise(R = portion_R(cipr, as_percent = TRUE),
I = portion_I(cipr, as_percent = TRUE),
S = portion_S(cipr, as_percent = TRUE),
n = n_rsi(cipr),
2018-08-03 14:49:29 +02:00
total = n()) %>%
pull(n) %>%
sum(),
1404)
2018-05-02 14:56:25 +02:00
# count of cases
expect_equal(septic_patients %>%
group_by(hospital_id) %>%
2018-08-10 15:01:05 +02:00
summarise(cipro_p = portion_S(cipr, as_percent = TRUE),
2018-05-02 14:56:25 +02:00
cipro_n = n_rsi(cipr),
2018-08-10 15:01:05 +02:00
genta_p = portion_S(gent, as_percent = TRUE),
2018-05-02 14:56:25 +02:00
genta_n = n_rsi(gent),
2018-08-10 15:01:05 +02:00
combination_p = portion_S(cipr, gent, as_percent = TRUE),
combination_n = n_rsi(cipr, gent)) %>%
2018-05-02 14:56:25 +02:00
pull(combination_n),
c(202, 482, 201, 499))
2018-07-17 10:32:26 +02:00
2018-08-10 15:01:05 +02:00
expect_warning(portion_R(as.character(septic_patients$amcl)))
expect_warning(portion_S(as.character(septic_patients$amcl)))
expect_warning(portion_S(as.character(septic_patients$amcl,
2018-07-17 10:32:26 +02:00
septic_patients$gent)))
2018-08-12 17:44:06 +02:00
expect_equal(n_rsi(as.character(septic_patients$amcl,
septic_patients$gent)),
1570)
2018-07-17 10:32:26 +02:00
2018-07-30 00:57:49 +02:00
# check for errors
2018-08-10 15:01:05 +02:00
expect_error(portion_IR(septic_patients %>% select(amox, amcl)))
expect_error(portion_IR("test", minimum = "test"))
expect_error(portion_IR("test", as_percent = "test"))
expect_error(portion_I(septic_patients %>% select(amox, amcl)))
expect_error(portion_I("test", minimum = "test"))
expect_error(portion_I("test", as_percent = "test"))
expect_error(portion_S("test", minimum = "test"))
expect_error(portion_S("test", as_percent = "test"))
expect_error(portion_S(septic_patients %>% select(amox, amcl)))
expect_error(portion_S("R", septic_patients %>% select(amox, amcl)))
2018-08-12 17:44:06 +02:00
expect_error(n_rsi(septic_patients %>% select(amox, amcl)))
expect_error(n_rsi(septic_patients$amox, septic_patients %>% select(amox, amcl)))
2018-07-30 00:57:49 +02:00
# check too low amount of isolates
2018-08-10 15:01:05 +02:00
expect_identical(portion_R(septic_patients$amox, minimum = nrow(septic_patients) + 1),
2018-08-03 14:49:29 +02:00
NA)
2018-08-10 15:01:05 +02:00
expect_identical(portion_I(septic_patients$amox, minimum = nrow(septic_patients) + 1),
2018-07-30 00:57:49 +02:00
NA)
2018-08-10 15:01:05 +02:00
expect_identical(portion_S(septic_patients$amox, minimum = nrow(septic_patients) + 1),
2018-07-30 00:57:49 +02:00
NA)
# warning for speed loss
2018-08-10 15:01:05 +02:00
expect_warning(portion_R(as.character(septic_patients$gent)))
expect_warning(portion_I(as.character(septic_patients$gent)))
expect_warning(portion_S(septic_patients$amcl, as.character(septic_patients$gent)))
2018-07-30 00:57:49 +02:00
2018-03-27 17:43:42 +02:00
})
2018-07-17 11:18:36 +02:00
test_that("old rsi works", {
# amox resistance in `septic_patients` should be around 66.33%
2018-07-29 22:14:51 +02:00
expect_equal(suppressWarnings(rsi(septic_patients$amox)), 0.6633, tolerance = 0.0001)
expect_equal(suppressWarnings(rsi(septic_patients$amox, interpretation = "S")), 1 - 0.6633, tolerance = 0.0001)
2018-08-10 15:01:05 +02:00
2018-07-17 11:18:36 +02:00
# pita+genta susceptibility around 98.09%
2018-07-29 22:14:51 +02:00
expect_equal(suppressWarnings(rsi(septic_patients$pita,
septic_patients$gent,
interpretation = "S",
info = TRUE)),
0.9535,
2018-07-17 11:18:36 +02:00
tolerance = 0.0001)
# count of cases
expect_equal(septic_patients %>%
group_by(hospital_id) %>%
2018-07-29 22:14:51 +02:00
summarise(cipro_S = suppressWarnings(rsi(cipr, interpretation = "S",
as_percent = TRUE, warning = FALSE)),
2018-07-17 11:18:36 +02:00
cipro_n = n_rsi(cipr),
2018-07-29 22:14:51 +02:00
genta_S = suppressWarnings(rsi(gent, interpretation = "S",
as_percent = TRUE, warning = FALSE)),
2018-07-17 11:18:36 +02:00
genta_n = n_rsi(gent),
2018-07-29 22:14:51 +02:00
combination_S = suppressWarnings(rsi(cipr, gent, interpretation = "S",
as_percent = TRUE, warning = FALSE)),
2018-07-17 11:18:36 +02:00
combination_n = n_rsi(cipr, gent)) %>%
pull(combination_n),
c(202, 482, 201, 499))
2018-08-12 17:44:06 +02:00
# portion_df
expect_equal(
2018-08-13 16:42:37 +02:00
septic_patients %>% select(amox) %>% portion_df() %>% pull(Percentage),
2018-08-12 17:44:06 +02:00
c(septic_patients$amox %>% portion_S(),
septic_patients$amox %>% portion_I(),
septic_patients$amox %>% portion_R())
)
2018-07-17 11:18:36 +02:00
})
2018-03-27 17:43:42 +02:00
test_that("prediction of rsi works", {
2018-04-03 11:08:31 +02:00
amox_R <- septic_patients %>%
filter(bactid == "ESCCOL") %>%
rsi_predict(col_ab = "amox",
col_date = "date",
2018-07-26 16:30:42 +02:00
minimum = 10,
2018-07-17 11:18:36 +02:00
info = TRUE) %>%
2018-07-28 09:34:03 +02:00
pull("value")
# amox resistance will increase according to data set `septic_patients`
expect_true(amox_R[3] < amox_R[20])
2018-04-03 11:08:31 +02:00
expect_output(rsi_predict(tbl = filter(septic_patients, bactid == "ESCCOL"),
model = "binomial",
col_ab = "amox",
col_date = "date",
info = TRUE))
expect_output(rsi_predict(tbl = filter(septic_patients, bactid == "ESCCOL"),
model = "loglin",
col_ab = "amox",
col_date = "date",
info = TRUE))
expect_output(rsi_predict(tbl = filter(septic_patients, bactid == "ESCCOL"),
model = "lin",
col_ab = "amox",
col_date = "date",
info = TRUE))
expect_error(rsi_predict(tbl = filter(septic_patients, bactid == "ESCCOL"),
2018-03-27 17:43:42 +02:00
model = "INVALID MODEL",
col_ab = "amox",
col_date = "date",
2018-07-17 11:18:36 +02:00
info = TRUE))
2018-04-03 11:08:31 +02:00
expect_error(rsi_predict(tbl = filter(septic_patients, bactid == "ESCCOL"),
2018-03-27 17:43:42 +02:00
col_ab = "NOT EXISTING COLUMN",
col_date = "date",
2018-07-17 11:18:36 +02:00
info = TRUE))
2018-04-03 11:08:31 +02:00
expect_error(rsi_predict(tbl = filter(septic_patients, bactid == "ESCCOL"),
2018-03-27 17:43:42 +02:00
col_ab = "amox",
col_date = "NOT EXISTING COLUMN",
2018-07-17 11:18:36 +02:00
info = TRUE))
2018-07-30 00:57:49 +02:00
# almost all E. coli are mero S in the Netherlands :)
expect_error(resistance_predict(tbl = filter(septic_patients, bactid == "ESCCOL"),
col_ab = "mero",
col_date = "date",
info = TRUE))
2018-03-27 17:43:42 +02:00
})