tests/testthat/test-aggregate_simulation_results.R

In lmiratrix/simITS: Analysis via Simulation of Interrupted Time Series (ITS) Data

context("test-summarization-testing")


test_that("Aggregation of simulation results works", {
  
  data( "mecklenberg")
  t0 = 0
  preds = process_outcome_model( outcomename = "pbail", timename = "month", dat=mecklenberg,
                                 t0=t0, R = 10,
                                 summarize = FALSE, smooth=FALSE )
  
  sr =  aggregate_simulation_results( mecklenberg, preds, outcomename="pbail", timename = "month" )
  sr
  
  expect_equal( length( sr ), 2 )
  expect_equal( length( sr$t ), 10 )
})



test_that("Alternate function works for aggregation", {
  
  agr_2 = function( res, outcomename,
                    months = 1:54,
                    ... ) {
    
    mts = dplyr::filter( res, month %in% months )
    median( mts[[outcomename]] )
  }
  
    
    
  data( "mecklenberg")
  t0 = 0
  preds = process_outcome_model( outcomename= "pbail", timename="month", dat= mecklenberg,
                                 t0=t0, R = 10,
                                 summarize = FALSE, smooth=FALSE )
  
  sr =  aggregate_simulation_results( mecklenberg, preds, outcomename="pbail", timename="month",
                                      summarizer = agr_2, months=1:5 )
  sr
  
  expect_equal( length( sr ), 2 )
  expect_equal( length( sr$t ), 10 )
})