impelmented test architecture for comparing weight_method to manual calculation

Author: awinterstetter

R/PipeOpClassWeightsEx.R

@@ -114,7 +114,7 @@ PipeOpClassWeightsEx = R6Class("PipeOpClassWeightsEx",
 
       truth = task$truth()
 
-      class_frequency = table(truth) / length(truth)
+      class_frequency = prop.table(table(truth))
       class_names = names(class_frequency)
 
       weights_by_class = switch(pv$weight_method,

tests/testthat/test_pipeop_classweightsex.R

@@ -43,8 +43,18 @@ test_that("PipeOpClassWeightsEx", {
   nt = poicf$train(list(task))[[1L]]
   expect_equal(nt$data(), task$data())
 
+  freq = prop.table(table(task$truth()))
+  manual_weights = 1 / freq[task$truth()]
+
+  if ("weights_learner" %in% names(nt$col_roles)) {
+    computed_weights = nt$weights_learner
+  } else {
+    computed_weights = nt$weights
+  }
+
+  expect_equal(computed_weights[["weight"]], as.numeric(unclass(manual_weights)))
+
 
-  # manual_weights = as.data.table(1 / table(task$data()$Species))
   # weights = if ("weights_learner" %in% names(nt)) "weights_learner" else "weights"
   # expect_equal(nt[[weights]]$weight, ifelse(nt$truth(nt[[weights]]$row_ids) == "neg", 1, 3))
 })