def test_coef_md(self):
blim_md = BLIM(self.k, self.n_r)
beta, eta, p_k = blim_md.coef()
self.assertListEqual(list(blim_md.beta), list(beta))
self.assertListEqual(blim_md.beta.values.tolist(),
beta.values.tolist())
self.assertListEqual(list(blim_md.eta), list(eta))
self.assertListEqual(blim_md.eta.values.tolist(), eta.values.tolist())
self.assertListEqual(list(blim_md.p_k), list(p_k))
self.assertListEqual(blim_md.p_k.values.tolist(), p_k.values.tolist())
def test_coef_ml(self):
blim_ml = BLIM(self.k, self.n_r, method="ML")
beta, eta, p_k = blim_ml.coef()
self.assertListEqual(list(blim_ml.beta), list(beta))
self.assertListEqual(blim_ml.beta.values.tolist(),
beta.values.tolist())
self.assertListEqual(list(blim_ml.eta), list(eta))
self.assertListEqual(blim_ml.eta.values.tolist(), eta.values.tolist())
self.assertListEqual(list(blim_ml.p_k), list(p_k))
self.assertListEqual(blim_ml.p_k.values.tolist(), p_k.values.tolist())