def test_pars_total_indexes(): pars_oi = ["mu", "tau", "eta", "theta", "lp__"] dims_oi = [[], [], [8], [8], []] fnames_oi = [ "mu", "tau", "eta[1]", "eta[2]", "eta[3]", "eta[4]", "eta[5]", "eta[6]", "eta[7]", "eta[8]", "theta[1]", "theta[2]", "theta[3]", "theta[4]", "theta[5]", "theta[6]", "theta[7]", "theta[8]", "lp__", ] pars = ["mu", "tau", "eta", "theta", "lp__"] rslt = _pars_total_indexes(pars_oi, dims_oi, fnames_oi, pars) assert rslt["mu"] == (0,) assert rslt["tau"] == (1,) assert rslt["eta"] == (2, 3, 4, 5, 6, 7, 8, 9) assert rslt["theta"] == (10, 11, 12, 13, 14, 15, 16, 17) assert rslt["lp__"] == (18,)
def test_pars_total_indexes(): pars_oi = ['mu', 'tau', 'eta', 'theta', 'lp__'] dims_oi = [[], [], [8], [8], []] fnames_oi = ['mu', 'tau', 'eta[1]', 'eta[2]', 'eta[3]', 'eta[4]', 'eta[5]', 'eta[6]', 'eta[7]', 'eta[8]', 'theta[1]', 'theta[2]', 'theta[3]', 'theta[4]', 'theta[5]', 'theta[6]', 'theta[7]', 'theta[8]', 'lp__'] pars = ['mu', 'tau', 'eta', 'theta', 'lp__'] rslt = misc._pars_total_indexes(pars_oi, dims_oi, fnames_oi, pars) assert rslt['mu'] == (0,) assert rslt['tau'] == (1,) assert rslt['eta'] == (2, 3, 4, 5, 6, 7, 8, 9) assert rslt['theta'] == (10, 11, 12, 13, 14, 15, 16, 17) assert rslt['lp__'] == (18,)
def test_pars_total_indexes(self): pars_oi = ['mu', 'tau', 'eta', 'theta', 'lp__'] dims_oi = [[], [], [8], [8], []] fnames_oi = [ 'mu', 'tau', 'eta[1]', 'eta[2]', 'eta[3]', 'eta[4]', 'eta[5]', 'eta[6]', 'eta[7]', 'eta[8]', 'theta[1]', 'theta[2]', 'theta[3]', 'theta[4]', 'theta[5]', 'theta[6]', 'theta[7]', 'theta[8]', 'lp__' ] pars = ['mu', 'tau', 'eta', 'theta', 'lp__'] rslt = misc._pars_total_indexes(pars_oi, dims_oi, fnames_oi, pars) self.assertEqual(rslt['mu'], (0, )) self.assertEqual(rslt['tau'], (1, )) self.assertEqual(rslt['eta'], (2, 3, 4, 5, 6, 7, 8, 9)) self.assertEqual(rslt['theta'], (10, 11, 12, 13, 14, 15, 16, 17)) self.assertEqual(rslt['lp__'], (18, ))
def test_pars_total_indexes(): pars_oi = ['mu', 'tau', 'eta', 'theta', 'lp__'] dims_oi = [[], [], [8], [8], []] fnames_oi = [ 'mu', 'tau', 'eta[1]', 'eta[2]', 'eta[3]', 'eta[4]', 'eta[5]', 'eta[6]', 'eta[7]', 'eta[8]', 'theta[1]', 'theta[2]', 'theta[3]', 'theta[4]', 'theta[5]', 'theta[6]', 'theta[7]', 'theta[8]', 'lp__' ] pars = ['mu', 'tau', 'eta', 'theta', 'lp__'] rslt = _pars_total_indexes(pars_oi, dims_oi, fnames_oi, pars) assert rslt['mu'] == (0, ) assert rslt['tau'] == (1, ) assert rslt['eta'] == (2, 3, 4, 5, 6, 7, 8, 9) assert rslt['theta'] == (10, 11, 12, 13, 14, 15, 16, 17) assert rslt['lp__'] == (18, )