tests/testthat/test-woebin.R

In scorecard: Credit Risk Scorecard

# http://r-pkgs.had.co.nz/tests.html
# https://testthat.r-lib.org/
library(scorecard)
library(data.table)
data("germancredit")
setDT(germancredit)
xcols = c("status.of.existing.checking.account", "duration.in.month", "credit.history", "purpose")
ycol = 'creditability'
dtxy = germancredit[, c(ycol, xcols), with = FALSE]

test_that('woebin, woebin_ply', {
  # y provided
  for (m1 in c('tree', 'chimerge', 'freq', 'width')) {
    bin1 = woebin(germancredit, y = ycol, x = xcols, method = m1, print_info=F)
    expect_true(inherits(bin1, 'list'))
    expect_true(inherits(bin1[[1]], 'data.frame'))
    expect_true(inherits(rbindlist(bin1), 'data.frame'))

    dat1_woe = woebin_ply(dtxy, bin1, print_info=F)
    expect_true(inherits(dat1_woe, 'data.frame'))
    dat2_woe = woebin_ply(germancredit, rbindlist(bin1), print_info=F)
    expect_true(inherits(dat2_woe, 'data.frame'))
  }

  # y is not provided when method is in equal freq or width
  for (m2 in c('freq', 'width')) {
    bin2  = woebin(germancredit, y=NULL, x=xcols, print_info=F, method = m2)
    expect_true(inherits(bin2, 'list'))
    expect_true(inherits(bin2[[1]], 'data.frame'))
    expect_true(inherits(rbindlist(bin2), 'data.frame'))

    dat1_bin = woebin_ply(dtxy, bin2, to = 'bin', print_info=F)
    expect_true(inherits(dat1_bin, 'data.frame'))
    dat2_bin = woebin_ply(germancredit, rbindlist(bin2), to = 'bin', print_info=F)
    expect_true(inherits(dat2_bin, 'data.frame'))
  }


})