def test_normal_mean(self):
mean = distribution_generator(500, "normal", mean=5).mean()
print(mean)
self.assertTrue(4 <= mean <= 6)
def test_error(self):
with self.assertRaises(ValueError):
distribution_generator(500, "cat")
def test_poisson(self):
output = distribution_generator(500, "poisson", interval=20).mean()
print(output)
self.assertTrue(19 <= output <= 21)
def test_binomial(self):
output = distribution_generator(500, "binomial", n=1, p=0.75).mean()
print(output)
self.assertTrue(0.7 <= output <= 0.8)
def test_normal_sd(self):
sd = distribution_generator(500, "normal", sd=10).std()
print(sd)
self.assertTrue(9 <= sd <= 11)