tests/testthat/test-fitBR.R

In sverchkov/BaselineRegularization: Baseline Regularization

context("fitBR")

test_that("Regularized mini interval-tying single-patient model fit", {

  # Tiny toy dataset:
  # * 1 patients, with 14 intervals
  # * 2 drugs,
  # Interval length: 5 2 1 5 10 10 5 10 4 7 3 12 20 12
  # ADE:             0 0 1 0  1  0 0  0 0 1 0  0  0  0
  # Drug1:           0 1 1 1  1  0 0  0 1 1 1  1  0  0
  # Drug2:           1 1 1 1  1  1 0  0 0 0 0  0  0  0

  X <- Matrix::Matrix( c(
    0, 1,
    1, 1,
    1, 1,
    1, 1,
    1, 1,
    0, 1,
    0, 0,
    0, 0,
    1, 0,
    1, 0,
    1, 0,
    1, 0,
    0, 0,
    0, 0
    ), nrow=14, ncol=2 )
  interval_baseline_parameter <- 1:14
  l <- c( 5, 2, 1, 5, 10, 10, 5, 10, 4, 7, 3, 12, 20, 12 )
  n <- c( 0, 0, 1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0 )
  baseline_parameter_obs_period <- rep( 1, 14 )

  fit = fitBR( interval_baseline_parameter, baseline_parameter_obs_period, X, l, n, lambda1 = 0.01, lambda2=0.05, lambda3 = 0 )

  expect_equal( as.numeric( fit$beta ), c( 0.5286806212, 0.6699739 ) )
})