def test_variable_lv(self):
# pylint: disable=C0103
# construct fit using existing sampler output
exe = os.path.join(DATAFILES_PATH, 'lotka-volterra' + EXTENSION)
jdata = os.path.join(DATAFILES_PATH, 'lotka-volterra.data.json')
sampler_args = SamplerArgs(iter_sampling=20)
cmdstan_args = CmdStanArgs(
model_name='lotka-volterra',
model_exe=exe,
chain_ids=[1],
seed=12345,
data=jdata,
output_dir=DATAFILES_PATH,
method_args=sampler_args,
)
runset = RunSet(args=cmdstan_args, chains=1)
runset._csv_files = [
os.path.join(DATAFILES_PATH, 'lotka-volterra.csv')
]
runset._set_retcode(0, 0)
fit = CmdStanMCMC(runset)
self.assertEqual(20, fit.num_draws)
self.assertEqual(8, len(fit._stan_variable_dims))
self.assertTrue('z' in fit._stan_variable_dims)
self.assertEqual(fit._stan_variable_dims['z'], (20, 2))
z = fit.stan_variable(name='z')
self.assertEqual(z.shape, (20, 20, 2))
theta = fit.stan_variable(name='theta')
self.assertEqual(theta.shape, (20, 4))
def test_variables_3d(self):
# construct fit using existing sampler output
exe = os.path.join(DATAFILES_PATH, 'multidim_vars' + EXTENSION)
jdata = os.path.join(DATAFILES_PATH, 'logistic.data.R')
sampler_args = SamplerArgs(iter_sampling=20)
cmdstan_args = CmdStanArgs(
model_name='multidim_vars',
model_exe=exe,
chain_ids=[1],
seed=12345,
data=jdata,
output_dir=DATAFILES_PATH,
method_args=sampler_args,
)
runset = RunSet(args=cmdstan_args, chains=1)
runset._csv_files = [os.path.join(DATAFILES_PATH, 'multidim_vars.csv')]
runset._set_retcode(0, 0)
fit = CmdStanMCMC(runset)
self.assertEqual(20, fit.num_draws_sampling)
self.assertEqual(3, len(fit.stan_vars_dims))
self.assertTrue('y_rep' in fit.stan_vars_dims)
self.assertEqual(fit.stan_vars_dims['y_rep'], (5, 4, 3))
var_y_rep = fit.stan_variable(name='y_rep')
self.assertEqual(var_y_rep.shape, (20, 5, 4, 3))
var_beta = fit.stan_variable(name='beta')
self.assertEqual(var_beta.shape, (20, 2))
var_frac_60 = fit.stan_variable(name='frac_60')
self.assertEqual(var_frac_60.shape, (20, ))
vars = fit.stan_variables()
self.assertEqual(len(vars), len(fit.stan_vars_dims))
self.assertTrue('y_rep' in vars)
self.assertEqual(vars['y_rep'].shape, (20, 5, 4, 3))
self.assertTrue('beta' in vars)
self.assertEqual(vars['beta'].shape, (20, 2))
self.assertTrue('frac_60' in vars)
self.assertEqual(vars['frac_60'].shape, (20, ))