def test_shannon(self): """shannon should match hand-calculated values""" c = array([5]) self.assertFloatEqual(shannon(c), 0) c = array([5,5]) self.assertFloatEqual(shannon(c), 1) c = array([1,1,1,1,0]) self.assertEqual(shannon(c), 2)
def test_heip_e(self): """heip e should match hand-calculated value""" c = array([1,2,3,1]) h = shannon(c, base=e) expected = exp(h-1)/3 self.assertEqual(heip_e(c), expected)