From 3b9781cf6436adc4cbb3cb0a21b8568d67c9dd25 Mon Sep 17 00:00:00 2001 From: Albert Julius Liu Date: Thu, 26 Dec 2024 23:58:33 -0800 Subject: [PATCH] add simple test for bound generator --- tests/multiset_function_test.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/multiset_function_test.py b/tests/multiset_function_test.py index 1bcfc577..42e446f8 100644 --- 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