def test_multiple_phi(self): vb = VariationalBayes() gamma = array([2.0, 2.0, 2.0]) beta = self.init_beta phi = vb.new_phi(gamma, beta, 0, 2) prop = 0.27711205238850234 normalizer = sum(x * prop for x in beta[:, 0]) / 2.0 self.assertAlmostEqual(phi[0], beta[0][0] * prop / normalizer) self.assertAlmostEqual(phi[1], beta[1][0] * prop / normalizer) self.assertAlmostEqual(phi[2], beta[2][0] * prop / normalizer)