Python Unittest Problem : Probability Calculator

Python
1 
def test_prob_experiment(self):
        hat = prob_calculator.Hat(blue=3,red=2,green=6)
        probability = prob_calculator.experiment(hat=hat, expected_balls={"blue":2,"green":1}, num_balls_drawn=4, num_experiments=1000)
        actual = probability
        expected = 0.272
        self.assertAlmostEqual(actual, expected, delta = 0.01, msg = 'Expected experiment method to return a different probability.')
        hat = prob_calculator.Hat(yellow=5,red=1,green=3,blue=9,test=1)
        probability = prob_calculator.experiment(hat=hat, expected_balls={"yellow":2,"blue":3,"test":1}, num_balls_drawn=20, num_experiments=100)
        actual = probability
        expected = 1.0
        self.assertAlmostEqual(actual, expected, delta = 0.01, msg = 'Expected experiment method to return a different probability.')

Hello! In the probability calculator project my code keeps failing the second part of this unittest (the one where probability is 1) because the test is trying to draw more balls than there are in the hat (only 19 in hat and its trying to draw 20). Is this a mistake with the unittest or am I supposed to account for such a situation in my code? Please help me with this.

2

You are supposed to account for that situation in your code :wink:

3

Gottit, thank you! :slight_smile:

4

This topic was automatically closed 182 days after the last reply. New replies are no longer allowed.