add test_map_and_time_extra_args

HighDiceRoller · Sep 15, 2024 · bb757b7 · bb757b7
1 parent 5c7b91b
commit bb757b7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/map_test.py b/tests/map_test.py
@@ -176,3 +176,10 @@ def test_fractional_coin():
 def test_stochastic_round():
     assert ((6 @ d6) / 2).stochastic_round().mean() == 10.5
     assert ((6 @ d6) / Fraction(3)).stochastic_round().mean() == 7
+
+
+def test_map_and_time_extra_args():
+    def test_function(current, roll):
+        return min(current + roll, 10)
+    result = Die([0]).map_and_time(test_function, d6, repeat=10)
+    assert result.marginals[1].mean() == d6.mean_time_to_sum(10)