Merge pull request #204 from rkdarst/python-yahtzee

yahtzee_sol: Test against a distribution
coderefinery · Aug 23, 2023 · 7d8b830 · 7d8b830
2 parents 3278ee3 + 8b5a7ce
commit 7d8b830
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/content/code/python/yahtzee_sol.py b/content/code/python/yahtzee_sol.py
@@ -5,15 +5,16 @@ def test_roll_dice():
     assert roll_dice(5) == [5, 2, 5, 2, 3]
 
 
+import pytest
 def test_yahtzee():
     random.seed(1)
-    num_games = 10000
+    n_tests = 1_000_000
 
     winning_games = list(
         filter(
             lambda x: x == 5,
-            [yahtzee() for _ in range(num_games)],
+            [yahtzee() for _ in range(n_tests)],
         )
     )
 
-    assert len(winning_games) == 460
+    assert len(winning_games) / n_tests == pytest.approx(0.046, abs=0.01)