# Python Unittest Problem : Probability Calculator

``````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.

You are supposed to account for that situation in your code

Gottit, thank you!

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