Skip to content

Commit 09f0ba0

Browse files
authored
test: use more specialised test functions (#1408)
1 parent 2e24f7f commit 09f0ba0

File tree

9 files changed

+27
-27
lines changed

9 files changed

+27
-27
lines changed

DESCRIPTION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ Suggests:
7575
remotes,
7676
RhpcBLASctl,
7777
rpart,
78-
testthat (>= 3.2.0)
78+
testthat (>= 3.3.0)
7979
Encoding: UTF-8
8080
Config/testthat/edition: 3
8181
Config/testthat/parallel: false

inst/testthat/helper_expectations.R

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ expect_backend = function(b) {
116116
checkmate::expect_list(b$distinct(rn, "_not_existing_"), len = 0L, names = "named")
117117
d = b$distinct(rn, c("_not_existing_", rev(cn), "_also_not_existing_"))
118118
checkmate::expect_list(d, names = "unique")
119-
testthat::expect_equal(names(d), rev(cn))
119+
checkmate::expect_names(names(d), identical.to = rev(cn))
120120

121121
d = b$distinct(rn1, cn)
122122
checkmate::expect_list(d, len = length(cn), names = "unique", any.missing = FALSE)
@@ -159,15 +159,15 @@ expect_iris_backend = function(b, n_missing = 0L) {
159159

160160
x = b$head(2)
161161
checkmate::expect_data_table(x, nrows = 2L, ncols = 6L, any.missing = FALSE)
162-
checkmate::expect_set_equal(names(x), c(names(iris), b$primary_key))
162+
checkmate::expect_names(names(x), permutation.of = c(names(iris), b$primary_key))
163163

164164
x = b$distinct(b$rownames, "Species")
165165
checkmate::expect_list(x, "character", len = 1)
166166
checkmate::expect_set_equal(x$Species, levels(iris$Species))
167167

168168
x = b$data(rows = 2:10, cols = c(b$primary_key, "Species", "Sepal.Width"))
169169
checkmate::expect_data_table(x, nrows = 9L, ncols = 3L)
170-
checkmate::expect_set_equal(names(x), c(b$primary_key, "Species", "Sepal.Width"))
170+
checkmate::expect_names(names(x), identical.to = c(b$primary_key, "Species", "Sepal.Width"))
171171
checkmate::expect_set_equal(x[[b$primary_key]], 2:10)
172172

173173
if (is.factor(x$Species)) {
@@ -693,11 +693,11 @@ expect_resultdata = function(rdata, consistency = TRUE) {
693693
data = rdata$data
694694

695695
proto = mlr3:::star_init()
696-
checkmate::expect_set_equal(names(data), names(proto))
696+
checkmate::expect_names(names(data), identical.to = names(proto))
697697

698698
for (nn in names(proto)) {
699699
checkmate::expect_data_table(data[[nn]], key = data.table::key(proto[[nn]]))
700-
testthat::expect_equal(names(data[[nn]]), names(proto[[nn]]))
700+
checkmate::expect_names(names(data[[nn]]), identical.to = names(proto[[nn]]))
701701
}
702702

703703
checkmate::expect_character(data$uhashes$uhash, unique = TRUE)

tests/testthat/test_DataBackendRename.R

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@ test_that("DataBackendRename", {
66
expect_backend(b)
77

88
expect_set_equal(b$colnames, c(new, "..row_id"))
9-
expect_set_equal(names(b$data(b$rownames, b$colnames)), c(new, "..row_id"))
10-
expect_set_equal(names(b$head()), c(new, "..row_id"))
9+
expect_names(names(b$data(b$rownames, b$colnames)), identical.to = c(new, "..row_id"))
10+
expect_names(names(b$head()), identical.to = c(new, "..row_id"))
1111

1212
expect_data_table(b$data(rows = b$rownames, cols = old), ncols = 0)
1313
expect_data_table(b$data(rows = b$rownames, cols = new), ncols = 5)
@@ -26,8 +26,8 @@ test_that("DataBackendRename / partial rename", {
2626
expected = c(map_values(names(iris), old, new), "..row_id")
2727
expect_set_equal(b$colnames, expected)
2828

29-
expect_set_equal(names(b$data(b$rownames, b$colnames)), expected)
30-
expect_set_equal(names(b$head()), expected)
29+
expect_names(names(b$data(b$rownames, b$colnames)), identical.to = expected)
30+
expect_names(names(b$head()), identical.to = expected)
3131

3232
expect_data_table(b$data(rows = b$rownames, cols = expected), ncols = 6)
3333

tests/testthat/test_Learner.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ test_that("learner timings", {
4949
learner = lrn("regr.rpart")
5050
t = learner$timings
5151
expect_equal(unname(t), as.double(c(NA, NA)))
52-
expect_equal(names(t), c("train", "predict"))
52+
expect_names(names(t), identical.to = c("train", "predict"))
5353

5454

5555
learner$train(tsk("mtcars"))

tests/testthat/test_PredictionClassif.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ test_that("confusion", {
109109
expect_matrix(cm, nrows = 3, ncols = 3, any.missing = FALSE)
110110
expect_equal(colnames(p$confusion), task$class_names)
111111
expect_equal(rownames(p$confusion), task$class_names)
112-
expect_equal(names(dimnames(cm)), c("response", "truth"))
112+
expect_names(names(dimnames(cm)), identical.to = c("response", "truth"))
113113
})
114114

115115
test_that("c", {

tests/testthat/test_Task.R

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ test_that("Feature columns can be reordered", {
44

55
task$col_roles$feature = new_order
66
expect_equal(task$feature_names, new_order)
7-
expect_equal(names(task$data(rows = 1)), c("median_house_value", new_order))
7+
expect_names(names(task$data(rows = 1)), identical.to = c("median_house_value", new_order))
88
})
99

1010
test_that("Task duplicates rows", {
@@ -781,12 +781,12 @@ test_that("rbind with weights", {
781781

782782
# Check combined weights
783783
combined_weights_lrn = task$weights_learner
784-
expect_equal(nrow(combined_weights_lrn), task$nrow)
784+
expect_shape(combined_weights_lrn, nrow = task$nrow)
785785
expect_equal(setkeyv(combined_weights_lrn[row_id %in% original_row_ids], "row_id"), original_weights_lrn)
786786
expect_equal(combined_weights_lrn[row_id %in% 151:160]$weight, 1001:1010)
787787

788788
combined_weights_msr = task$weights_measure
789-
expect_equal(nrow(combined_weights_msr), task$nrow)
789+
expect_shape(combined_weights_msr, nrow = task$nrow)
790790
# expect_equal(combined_weights_msr[row_id %in% original_row_ids], original_weights_msr) # ordering issue with join
791791
expect_equal(setkeyv(combined_weights_msr[list(original_row_ids), on = "row_id"], "row_id"), original_weights_msr)
792792
expect_equal(combined_weights_msr[row_id %in% 151:160]$weight, 2010:2001)
@@ -827,11 +827,11 @@ test_that("cbind with weights", {
827827
expect_true("weights_measure" %in% task$properties)
828828

829829
weights_lrn = task$weights_learner
830-
expect_equal(nrow(weights_lrn), task$nrow)
830+
expect_shape(weights_lrn, nrow = task$nrow)
831831
expect_equal(weights_lrn$weight, 1:150)
832832

833833
weights_msr = task$weights_measure
834-
expect_equal(nrow(weights_msr), task$nrow)
834+
expect_shape(weights_msr, nrow = task$nrow)
835835
expect_equal(weights_msr$weight, 150:1)
836836

837837
# Check that original features/target are still there

tests/testthat/test_benchmark.R

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -480,17 +480,17 @@ test_that("param_values in benchmark", {
480480
bmr = benchmark(design)
481481
expect_benchmark_result(bmr)
482482
expect_equal(bmr$n_resample_results, 1)
483-
expect_equal(nrow(as.data.table(bmr)), 3)
483+
expect_shape(as.data.table(bmr), nrow = 3L)
484484
learner = bmr$resample_result(1)$learner
485485
expect_equal(learner$param_set$values$x, 1)
486-
expect_equal(nrow(as.data.table(bmr)), 3)
486+
expect_shape(as.data.table(bmr), nrow = 3L)
487487

488488
# multiple parameters set via manual design
489489
design = data.table(task = tasks, learner = learners, resampling = resamplings, param_values = list(list(list(x = 1), list(x = 0.5))))
490490
bmr = benchmark(design)
491491
expect_benchmark_result(bmr)
492492
expect_equal(bmr$n_resample_results, 2)
493-
expect_equal(nrow(as.data.table(bmr)), 6)
493+
expect_shape(as.data.table(bmr), nrow = 6L)
494494
learner = bmr$resample_result(1)$learner
495495
expect_equal(learner$param_set$values$x, 1)
496496
learner = bmr$resample_result(2)$learner

tests/testthat/test_encapsulate.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ test_that("evaluate / single step", {
3232
expect_message(expect_warning(disable_encapsulation(learner)$predict(task, row_ids = 101:150)))
3333
log = learner$log[stage == "predict"]
3434
expect_data_table(log)
35-
expect_equal(nrow(log), 0)
35+
expect_shape(log, nrow = 0L)
3636

3737
p = enable_encapsulation(learner)$predict(task, row_ids = 101:150)
3838
log = learner$log[stage == "predict"]

tests/testthat/test_resample.R

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -31,18 +31,18 @@ test_that("resample with no or multiple measures", {
3131
for (measures in list(mlr_measures$mget(c("classif.ce", "classif.acc")), list())) {
3232
tab = rr$score(measures, ids = FALSE, predictions = TRUE)
3333
expect_data_table(tab, ncols = length(mlr_reflections$rr_names) + length(learner$predict_sets) + length(measures), nrows = 3L)
34-
expect_set_equal(names(tab), c(mlr_reflections$rr_names, ids(measures), paste0("prediction_", learner$predict_sets)))
34+
expect_names(names(tab), permutation.of = c(mlr_reflections$rr_names, ids(measures), paste0("prediction_", learner$predict_sets)))
3535
perf = rr$aggregate(measures)
3636
expect_numeric(perf, any.missing = FALSE, len = length(measures), names = "unique")
37-
expect_equal(names(perf), unname(ids(measures)))
37+
expect_names(names(perf), identical.to = unname(ids(measures)))
3838
}
3939
})
4040

4141
test_that("as_benchmark_result.ResampleResult", {
4242
measures = list(msr("classif.ce"), msr("classif.acc"))
4343
bmr = as_benchmark_result(rr)
4444
expect_benchmark_result(bmr)
45-
expect_equal(nrow(get_private(bmr)$.data), nrow(get_private(rr)$.data))
45+
expect_identical(nrow(get_private(bmr)$.data), nrow(get_private(rr)$.data))
4646
expect_set_equal(bmr$uhashes, rr$uhash)
4747
aggr = bmr$aggregate()
4848
expect_data_table(aggr, nrows = 1)
@@ -61,9 +61,9 @@ test_that("inputs are cloned", {
6161
})
6262

6363
test_that("memory footprint", {
64-
expect_equal(nrow(get_private(rr)$.data$data$learners), 1L)
65-
expect_equal(nrow(get_private(rr)$.data$data$tasks), 1L)
66-
expect_equal(nrow(get_private(rr)$.data$data$resamplings), 1L)
64+
expect_shape(get_private(rr)$.data$data$learners, nrow = 1L)
65+
expect_shape(get_private(rr)$.data$data$tasks, nrow = 1L)
66+
expect_shape(get_private(rr)$.data$data$resamplings, nrow = 1L)
6767
})
6868

6969
test_that("predict_type is checked", {

0 commit comments

Comments
 (0)