tests/testthat/test-select.R

In metrumresearchgroup/shredder: Rstan Manipulation API

testthat::context('select')
rats <- readRDS('../files/rats.Rds')
testthat::describe('names',{

  it('no pars',{
    testthat::expect_message(rats%>%stan_select(),regexp = 'no pars selected')
  })
    
  it('single par',{
    x <- rats%>%stan_select(mu_alpha)
    testthat::expect_equal(x@model_pars,'mu_alpha')
  })
  
  it('multiple pars',{
    x <- rats%>%stan_select(mu_alpha,mu_beta)
    testthat::expect_equal(x@model_pars,c('mu_alpha','mu_beta'))
  })
  
  it('par index',{
    x <- rats%>%stan_select(alpha[1])
    testthat::expect_equal(x@sim$fnames_oi,c('alpha[1]'))
  })

  it('character par',{
    x <- rats%>%stan_select('mu_alpha','mu_beta')
    testthat::expect_equal(x@model_pars,c('mu_alpha','mu_beta'))
  })  
    
  it('character par syms',{
    x <- rats%>%stan_select(!!! rlang::syms(c('mu_alpha','mu_beta')))
    testthat::expect_equal(x@model_pars,c('mu_alpha','mu_beta'))
  })

  it('regex character pars',{
    idx <- sprintf('alpha[%s]',1:5)
    x <- rats%>%stan_select(!!!rlang::syms(idx))
    testthat::expect_equal(x@sim$fnames_oi,idx)
  })
  

  it('remove summary',{
    rstan::summary(rats)
    x <- rats%>%stan_select(mu_alpha)
    testthat::expect_equal(x@model_pars,'mu_alpha')
  })
  
})

testthat::describe('partials',{
  
  it('no pars',{
    testthat::expect_message(rats%>%stan_select(stan_starts_with('mum')),regexp = 'no pars selected')
  })
  
  it('starts_with',{
    x <- rats%>%stan_select(stan_starts_with('mu'))
    testthat::expect_equal(x@model_pars,c('mu_alpha','mu_beta'))
  })

  it('ends_with',{
    x <- rats%>%stan_select(stan_ends_with('0'))
    testthat::expect_equal(x@model_pars,c('alpha0'))
  })
  
  it('starts_contains',{
    x <- rats%>%stan_select(stan_contains('alpha'))
    testthat::expect_equal(x@model_pars,c('alpha','mu_alpha','sigmasq_alpha','sigma_alpha','alpha0'))
  })
  
  it('mixed',{
    x <- rats%>%stan_select(alpha,stan_starts_with('mu'))
    testthat::expect_equal(x@model_pars,c('alpha','mu_alpha','mu_beta'))
  })
  
  it('par regex index',{
    x <- rats%>%stan_select(stan_contains('alpha\\[1\\]'))
    testthat::expect_equal(x@sim$fnames_oi,c('alpha[1]'))
  })
  
  it('par regex multiple index',{
    x <- rats%>%stan_select(stan_contains('alpha\\[[1-2]\\]'))
    testthat::expect_equal(x@sim$fnames_oi,c('alpha[1]','alpha[2]'))
  })
})