add simple test for bound generator

HighDiceRoller · Dec 27, 2024 · 3b9781c · 3b9781c
1 parent 0adac39
commit 3b9781c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/multiset_function_test.py b/tests/multiset_function_test.py
@@ -11,3 +11,12 @@ def evaluator(a):
         return a.sum()
 
     assert evaluator(d6.pool(3)) == 3 @ d6
+
+
+def test_bound_generator():
+
+    @multiset_function
+    def evaluator(a):
+        return (a + d6.pool(2)).sum()
+
+    assert evaluator(d6.pool(1)) == 3 @ d6