Coverage for lasso/math/test_stochastic.py: 100%

1import unittest 

2 

3from lasso.math.stochastic import jensen_shannon_entropy 

4 

5 

6class Test(unittest.TestCase): 

7 def test_jensen_shannon_entropy(self): 

8 p1 = [0.5, 0.5, 0.0] 

9 p2 = [0, 0.1, 0.9] 

10 

11 self.assertEqual(jensen_shannon_entropy(p1, p1), 0) 

12 self.assertAlmostEqual(jensen_shannon_entropy(p1, p2), 0.55797881790005399) 

13 self.assertAlmostEqual(jensen_shannon_entropy(p2, p1), 0.55797881790005399)