test_psi.py 文件源码

def test_gamma_equal_lambda():
    mu = np.linspace(-100, 100, 2)
    sigma = np.linspace(2, 200, 2)
    x = np.linspace(-200, 200, 3)
    lapse = np.linspace(0, 0.1, 4)
    guess = lapse
    psi = PsiMarginal.Psi(x, Pfunction='cGauss', nTrials=50, threshold=mu, thresholdPrior=('uniform', None),
                          slope=sigma, slopePrior=('uniform', None),
                          guessRate=guess, guessPrior=('uniform', None), lapseRate=lapse, lapsePrior=('uniform', None),
                          marginalize=True)
    assert psi.gammaEQlambda == True
    guess = np.array([0.5], dtype='float')
    psi2 = PsiMarginal.Psi(x, Pfunction='cGauss', nTrials=50, threshold=mu, thresholdPrior=('uniform', None),
                           slope=sigma, slopePrior=('uniform', None),
                           guessRate=guess, guessPrior=('uniform', None), lapseRate=lapse, lapsePrior=('uniform', None),
                           marginalize=True)
    assert psi2.gammaEQlambda == False