test: better object size estimate

Author: b-zhou

DESCRIPTION

@@ -41,7 +41,8 @@ Suggests:
     ranger,
     rpart,
     testthat (>= 3.0.0),
-    xgboost
+    xgboost,
+    pryr
 Remotes:
     catboost/catboost/catboost/R-package,
     mlr-org/mlr3,

tests/testthat/test_estimate_memory.R

@@ -11,7 +11,7 @@ task = as_task_classif(dt, target = "y")
 test_that("LearnerClassifRanger memory estimate is approximately correct", {
   learner = lrn("classif.ranger")
   learner$train(task)
-  true_mem = object.size(learner$model)
+  true_mem = pryr::object_size(learner$model)
   estim_mem = estimate_memory(learner, task)
 
   expect_true(true_mem <= estim_mem)
@@ -20,7 +20,7 @@ test_that("LearnerClassifRanger memory estimate is approximately correct", {
 test_that("LearnerClassifXgboost memory estimate is approximately correct", {
   learner = lrn("classif.xgboost")
   learner$train(task)
-  true_mem = object.size(learner$model)
+  true_mem = pryr::object_size(learner$model)
   estim_mem = estimate_memory(learner, task)
 
   expect_true(true_mem <= estim_mem)
@@ -29,7 +29,7 @@ test_that("LearnerClassifXgboost memory estimate is approximately correct", {
 test_that("LearnerClassifCatboost memory estimate is approximately correct", {
   learner = lrn("classif.catboost")
   learner$train(task)
-  true_mem = object.size(learner$model)
+  true_mem = pryr::object_size(learner$model)
   estim_mem = estimate_memory(learner, task)
 
   expect_true(true_mem <= estim_mem)
@@ -38,7 +38,7 @@ test_that("LearnerClassifCatboost memory estimate is approximately correct", {
 test_that("LearnerClassifLightGBM memory estimate is approximately correct", {
   learner = lrn("classif.lightgbm")
   learner$train(task)
-  true_mem = object.size(learner$model)
+  true_mem = pryr::object_size(learner$model)
   estim_mem = estimate_memory(learner, task)
 
   expect_true(true_mem <= estim_mem)
@@ -52,7 +52,7 @@ task = as_task_regr(dt, target = "y")
 test_that("LearnerRegrRanger memory estimate is approximately correct", {
   learner = lrn("regr.ranger")
   learner$train(task)
-  true_mem = object.size(learner$model)
+  true_mem = pryr::object_size(learner$model)
   estim_mem = estimate_memory(learner, task)
 
   expect_true(true_mem <= estim_mem)
@@ -61,7 +61,7 @@ test_that("LearnerRegrRanger memory estimate is approximately correct", {
 test_that("LearnerRegrXgboost memory estimate is approximately correct", {
   learner = lrn("regr.xgboost")
   learner$train(task)
-  true_mem = object.size(learner$model)
+  true_mem = pryr::object_size(learner$model)
   estim_mem = estimate_memory(learner, task)
 
   expect_true(true_mem <= estim_mem)
@@ -70,7 +70,7 @@ test_that("LearnerRegrXgboost memory estimate is approximately correct", {
 test_that("LearnerRegrCatboost memory estimate is approximately correct", {
   learner = lrn("regr.catboost")
   learner$train(task)
-  true_mem = object.size(learner$model)
+  true_mem = pryr::object_size(learner$model)
   estim_mem = estimate_memory(learner, task)
 
   expect_true(true_mem <= estim_mem)
@@ -79,7 +79,7 @@ test_that("LearnerRegrCatboost memory estimate is approximately correct", {
 test_that("LearnerRegrLightGBM memory estimate is approximately correct", {
   learner = lrn("regr.lightgbm")
   learner$train(task)
-  true_mem = object.size(learner$model)
+  true_mem = pryr::object_size(learner$model)
   estim_mem = estimate_memory(learner, task)
 
   expect_true(true_mem <= estim_mem)