def test_valid(self): d = Dist(2) self.assertFalse(d.valid()) d[0] = 2 self.assertTrue(d.valid()) d[1] = 2 self.assertTrue(d.valid()) d[0] = 0 self.assertTrue(d.valid()) d[1] = 0 self.assertFalse(d.valid())
def test_tick(self): d = Dist(2) self.assertEqual(1, d.tick(0)) self.assertEqual(2, d.tick(0)) self.assertEqual(2, d.counts()) self.assertTrue(d.valid())
def test_entropy_invalid_dist(self): d = Dist(5) self.assertFalse(d.valid()) self.assertTrue(isnan(entropy(d)))