Python _pars_total_indexes示例

示例#1

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,)

示例#2

文件： test_misc.py 项目： lazycrazyowl/pystan

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,)

示例#3

文件： test_misc.py 项目： aydinvardarr/flaskblog

 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, ))

示例#4

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, )