def test_at_least_more_than_1(self):
""" Compare our values against stattrek.com/online-calculator/hypergeometric.aspx """
assert abs(demo.probFunc(52,13,1,10) - 0.959813879728687) < 1e-6
assert abs(demo.probFunc(52,13,2,10) - 0.785674025219665) < 1e-6
assert abs(demo.probFunc(52,13,3,10) - 0.482333633494272) < 1e-6
assert abs(demo.probFunc(52,13,4,10) - 0.204271607745995) < 1e-6
assert abs(demo.probFunc(52,13,5,10) - 0.0568144728794855) < 1e-6
assert abs(demo.probFunc(52,13,6,10) - 0.00997514768659347) < 1e-6
assert abs(demo.probFunc(52,13,7,10) - 0.0010533714593759) < 1e-6
def test_drawCards_lt_numAtleast(self):
"""
If the number of cards drawn is less than the number of of successes at
least necessary then we expect 0 probability
"""
assert demo.probFunc(deckSize=52,numCardsInCategory=4,numAtLeast=1,numDraws=0) == 0
assert demo.probFunc(52,13,1,0) == 0
assert demo.probFunc(52,4,1,-1) == 0
assert demo.probFunc(52,13,1,-1) == 0
assert demo.probFunc(52,4,2,1) == 0
assert demo.probFunc(52,13,2,1) == 0
assert demo.probFunc(52,4,2,-1) == 0
assert demo.probFunc(52,13,2,-1) == 0
def test_drawcards_equal_totalCards(self):
assert demo.probFunc(52,4,1,52) == 1
assert demo.probFunc(52,13,1,52) == 1
def test_probFunc_equals_probFromZero(self):
assert abs(demo.probFunc(52,4,1,1) - demo.probFromZero(52,4,1)) < 1e-6
assert abs(demo.probFunc(52,13,1,3) - demo.probFromZero(52,13,3)) < 1e-6
assert abs(demo.probFunc(52,4,1,52) - demo.probFromZero(52,3,52)) < 1e-6
assert abs(demo.probFunc(52,13,1,52) - demo.probFromZero(52,3,52)) < 1e-6
def test_prob_1_ace_52_equals_1_minus_0_ace_in_52(self):
assert abs(demo.probFunc(52,4,1,1) - (1 - comb(48,1) / comb(52,1))) < 1e-6
def test_prob_1_ace_in_52_1_draw(self):
assert abs(demo.probFunc(52,4,1,1) - 4/52) < 1e-6
def test_prob_3_clubs_52_equals_1_minus_0_clubs_in_52(self):
assert abs(demo.probFunc(52,13,1,3) - (1 - comb(39,3) / comb(52, 3))) < 1e-6
def test_prob_three_draws_for_club_from_52_cards(self):
assert abs(demo.probFunc(52,13,1,3) - 0.586470588235294) < 1e-6