def test_add(self):
gaussian_one = Gaussian(25, 3)
gaussian_two = Gaussian(30, 4)
gaussian_sum = gaussian_one + gaussian_two
self.assertEqual(gaussian_sum.mean, 55)
self.assertEqual(gaussian_sum.stdev, 5)
def setUp(self):
self.gaussian = Gaussian(25, 2)
self.gaussian.read_data_file('probdists/numbers.txt')
class TestGaussianClass(unittest.TestCase):
def setUp(self):
self.gaussian = Gaussian(25, 2)
self.gaussian.read_data_file('probdists/numbers.txt')
def test_initialization(self):
self.assertEqual(self.gaussian.mean, 25, 'incorrect mean')
self.assertEqual(self.gaussian.stdev, 2,
'incorrect standard deviation')
def test_readdata(self):
self.assertEqual(self.gaussian.data,
[1, 3, 99, 100, 120, 32, 330, 23, 76, 44, 31],
'data not read in correctly')
def test_meancalculation(self):
self.gaussian.calculate_mean()
self.assertEqual(self.gaussian.mean,
sum(self.gaussian.data) /
float(len(self.gaussian.data)),
'calculated mean not as expected')
def test_stdevcalculation(self):
self.assertEqual(self.gaussian.calculate_stdev(),
92.87, 'sample standard deviation incorrect')
self.assertEqual(self.gaussian.calculate_stdev(False),
88.55, 'population standard deviation incorrect')
def test_cdf(self):
self.assertEqual(self.gaussian.calculate_cdf(25, 3), 0.500,
'calculate_cdf function does not give expected result')
self.gaussian.calculate_mean()
self.gaussian.calculate_stdev()
self.assertEqual(self.gaussian.calculate_cdf(75, 3), 0.487,
'calculate_cdf function after calculating mean and \
stdev does not give expected result')
def test_pdf(self):
self.assertEqual(self.gaussian.calculate_pdf(25, 5), 0.19947,
'calculate_pdf function does not give expected result')
self.gaussian.calculate_mean()
self.gaussian.calculate_stdev()
self.assertEqual(self.gaussian.calculate_pdf(75, 5), 0.00429,
'calculate_pdf function after calculating mean and \
stdev does not give expected result')
def test_add(self):
gaussian_one = Gaussian(25, 3)
gaussian_two = Gaussian(30, 4)
gaussian_sum = gaussian_one + gaussian_two
self.assertEqual(gaussian_sum.mean, 55)
self.assertEqual(gaussian_sum.stdev, 5)