def test_m(self):
vb = VariationalBayes()
vb.init([], "stuck", 3)
# vb.m_step(self.init_beta)
# self.assertAlmostEqual(self.init_beta[2][3], vb._beta[2][3])
topic_count = array([[5., 4., 3., 2., 1.], [0., 2., 2., 4., 1.],
[1., 1., 1., 1., 1.]])
vb.m_step(topic_count)
self.assertAlmostEqual(vb._beta[2][3], .2)
self.assertAlmostEqual(vb._beta[0][0], .33333333)
self.assertAlmostEqual(vb._beta[1][4], .11111111)
self.assertAlmostEqual(vb._beta[0][3], .13333333)
def test_m(self):
vb = VariationalBayes()
vb.init([], "stuck", 3)
# vb.m_step(self.init_beta)
# self.assertAlmostEqual(self.init_beta[2][3], vb._beta[2][3])
topic_count = array([[5., 4., 3., 2., 1.],
[0., 2., 2., 4., 1.],
[1., 1., 1., 1., 1.]])
vb.m_step(topic_count)
self.assertAlmostEqual(vb._beta[2][3], .2)
self.assertAlmostEqual(vb._beta[0][0], .33333333)
self.assertAlmostEqual(vb._beta[1][4], .11111111)
self.assertAlmostEqual(vb._beta[0][3], .13333333)
