def test_median(self):
rhtype = RateHetType.GAMMA_EQ_CAT_MEDIAN
rhm = RateHetManager(shape=0.5, n_cat=4, distrib_code=rhtype)
expected = [0.02907775, 0.28071453, 0.92477307, 2.76543465]
assert_list_eq(self, rhm.rates, expected)
rhm.set_shape(4.2)
expected = [0.49655820, 0.79929783, 1.10263121, 1.60151276]
assert_list_eq(self, rhm.rates, expected)
def test_mean(self):
rhtype = RateHetType.GAMMA_EQ_CAT_MEAN
rhm = RateHetManager(shape=0.5, n_cat=4, distrib_code=rhtype)
expected = [0.03338775, 0.25191592, 0.82026848, 2.89442785]
assert_list_eq(self, rhm.rates, expected)
self.assertEqual(rhm.n_cat, 4)
self.assertTrue(abs(rhm.shape-0.5) < 1.0e-5)
rhm.set_shape(4.2)
expected = [0.46502269, 0.78185321, 1.08432009, 1.66880400]
assert_list_eq(self, rhm.rates, expected)
self.assertEqual(rhm.n_cat, 4)
self.assertTrue(abs(rhm.shape-4.2) < 1.0e-5)
rhm = RateHetManager(shape=0.53, n_cat=10, distrib_code=rhtype)
expected = [0.00680095, 0.04407109, 0.11615143, 0.22625412, 0.38203227,
0.59769745, 0.90010946, 1.34639481, 2.09367223, 4.28681618]
assert_list_eq(self, rhm.rates, expected)
self.assertEqual(rhm.n_cat, 10)
self.assertTrue(abs(rhm.shape-0.53) < 1.0e-5)
