def test_probability_4(self):
expected = {0: 0.0625, 1: 0.25, 2: 0.375, 3: 0.25, 4: 0.0625}
self.assertEqual(probability(4), expected)
def test_probability_100(self):
expected = 0.08
self.assertAlmostEqual(probability(100)[50], expected, places=2)
def test_probability_1(self):
expected = {0: 0.5, 1: 0.5}
self.assertEqual(probability(1), expected)