def test_likelihood(self):
degree_list = np.array([5, 5, 5, 5], dtype=np.float)
lambda_estimate = degreefit.geometric_lambda_estimation(degree_list)
result = degreefit.geometric_likelihood_term(
lambda_estimate, degree_list)
self.assertEqual(len(degree_list), len(result))
for r in result:
self.assertAlmostEqual(0.0913, r, places=3)
def test_geometric_lambda_estimation(self):
degree_list = np.array([5, 5, 5, 5], dtype=np.float)
self.assertAlmostEqual(
4.48, degreefit.geometric_lambda_estimation(degree_list), places=2)
