tests/testthat/test-extract_draws.R

In bdrc: Bayesian Discharge Rating Curves

context('Advanced data wrangling of bdrc model objects')

check_spread_draws_MCMC_mat_concordance <- function(mod,dat){
    components <- unique(names(dat)[!(names(dat) %in% c('chain','iter','h'))])
    for(x in components){
        MCMC_output <- mod[[paste0(x,'_posterior')]]
        if(is.null(dim(MCMC_output))){
            if('h' %in% names(dat)){
                expect_equal(dat[[x]][dat$h==dat$h[1]],MCMC_output)
            }else{
                expect_equal(dat[[x]],MCMC_output)
            }
        }else{
            expect_equal(dat[[x]],c(t(MCMC_output)))
        }
    }
}

check_gather_draws_MCMC_mat_concordance <- function(mod,dat){
    components <- unique(dat$name)
    for(x in components){
        MCMC_output <- mod[[paste0(x,'_posterior')]]
        curr_dat <- dat[dat$name==x,]
        if(is.null(dim(MCMC_output))){
            if('h' %in% names(curr_dat)){
                expect_equal(curr_dat$value[curr_dat$h==curr_dat$h[1]],MCMC_output)
            }else{
                expect_equal(curr_dat$value,MCMC_output)
            }
        }else{
            expect_equal(curr_dat$value,c(t(MCMC_output)))
        }
    }
}

test_that("check different inputs to spread draws and gather draws", {
    skip_on_cran()
    expect_equal(spread_draws(plm0.fit,'c','sigma_eps'),spread_draws(plm0.fit,c('c','sigma_eps')))
    expect_equal(spread_draws(plm.fit,'c',paste0('eta_',1:3)),spread_draws(plm.fit,c('c','c',paste0('eta_',1:3))))
    expect_equal(spread_draws(gplm0.fit,'c','c'),spread_draws(gplm0.fit,'c'))
    expect_equal(spread_draws(gplm.fit,c('c','sigma_beta'),c('eta_1','sigma_eta')),spread_draws(gplm.fit,'c','sigma_beta','eta_1','sigma_eta'))
    expect_error(spread_draws(gplm.fit_known_c,'c'))

    expect_equal(gather_draws(plm0.fit,'c','sigma_eps'),gather_draws(plm0.fit,c('c','sigma_eps')))
    expect_equal(gather_draws(plm.fit,'c',paste0('eta_',1:3)),gather_draws(plm.fit,c('c','c',paste0('eta_',1:3))))
    expect_equal(gather_draws(gplm0.fit,'c','c'),gather_draws(gplm0.fit,'c'))
    expect_equal(gather_draws(gplm.fit,c('c','sigma_beta'),c('eta_1','sigma_eta')),gather_draws(gplm.fit,'c','sigma_beta','eta_1','sigma_eta'))
    expect_error(gather_draws(gplm.fit_known_c,'c'))
})

test_that("spread draws and gather draws are concordant with MCMC matrices", {
    skip_on_cran()
    #spread_draws
    check_spread_draws_MCMC_mat_concordance(plm0.fit,spread_draws(plm0.fit,'c','sigma_eps'))
    check_spread_draws_MCMC_mat_concordance(plm0.fit,spread_draws(plm0.fit,'rating_curve','rating_curve_mean','c'))

    check_spread_draws_MCMC_mat_concordance(plm.fit,spread_draws(plm.fit,'c',paste0('eta_',1:6)))
    check_spread_draws_MCMC_mat_concordance(plm.fit,spread_draws(plm.fit,'rating_curve','rating_curve_mean','sigma_eps','sigma_eta'))

    check_spread_draws_MCMC_mat_concordance(gplm0.fit,spread_draws(gplm0.fit,'c','phi_beta','sigma_beta'))
    check_spread_draws_MCMC_mat_concordance(gplm0.fit,spread_draws(gplm0.fit,'rating_curve','rating_curve_mean','beta','f','sigma_eps'))

    check_spread_draws_MCMC_mat_concordance(gplm.fit,spread_draws(gplm.fit,'c',paste0('eta_',1:6)))
    check_spread_draws_MCMC_mat_concordance(gplm.fit,spread_draws(gplm.fit,'rating_curve','rating_curve_mean','sigma_eps','beta','f','sigma_eta'))
    #gather_draws
    check_gather_draws_MCMC_mat_concordance(plm0.fit,spread_draws(plm0.fit,'c','sigma_eps'))
    check_gather_draws_MCMC_mat_concordance(plm0.fit,spread_draws(plm0.fit,'rating_curve','rating_curve_mean','c'))

    check_gather_draws_MCMC_mat_concordance(plm.fit,spread_draws(plm.fit,'c',paste0('eta_',1:6)))
    check_gather_draws_MCMC_mat_concordance(plm.fit,spread_draws(plm.fit,'rating_curve','rating_curve_mean','sigma_eps','sigma_eta'))

    check_gather_draws_MCMC_mat_concordance(gplm0.fit,spread_draws(gplm0.fit,'c','phi_beta','sigma_beta'))
    check_gather_draws_MCMC_mat_concordance(gplm0.fit,spread_draws(gplm0.fit,'rating_curve','rating_curve_mean','beta','f','sigma_eps'))

    check_gather_draws_MCMC_mat_concordance(gplm.fit,spread_draws(gplm.fit,'c',paste0('eta_',1:6)))
    check_gather_draws_MCMC_mat_concordance(gplm.fit,spread_draws(gplm.fit,'rating_curve','rating_curve_mean','sigma_eps','beta','f','sigma_eta'))
})

## TODO test transformed