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)