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)