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