[R-package] add assertions to test on lightgbm() weights (#5110)

microsoft · Mar 31, 2022 · 3ed0027 · 3ed0027
1 parent 60244e4
commit 3ed0027
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/R-package/tests/testthat/test_basic.R b/R-package/tests/testthat/test_basic.R
@@ -2941,6 +2941,7 @@ test_that("lightgbm() accepts 'weight' and 'weights'", {
     , nrounds = 5L
     , verbose = -1L
   )
+  expect_equal(model$.__enclos_env__$private$train_set$get_field("weight"), w)
 
   # Avoid a bad CRAN check due to partial argument matches
   lgb_args <- list(
@@ -2952,4 +2953,5 @@ test_that("lightgbm() accepts 'weight' and 'weights'", {
     , verbose = -1L
   )
   model <- do.call(lightgbm, lgb_args)
+  expect_equal(model$.__enclos_env__$private$train_set$get_field("weight"), w)
 })