def test_choices(self): self.assertEqual(choices(7, 0), 1) self.assertEqual(choices(7, 1), 7) self.assertEqual(choices(7, 5), 21)
def binom(p, n, k): return choices(n, k) * p ** k * (1 - p) ** (n - k)