Skip to content

Commit 1dea131

Browse files
committed
test: cover weighted cond fallback branch for 100% coverage
1 parent e1b4e2b commit 1dea131

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

tests/test_weighted_transitions.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -446,3 +446,13 @@ def test_cond_name_with_fractional_percentage(self):
446446
group = _WeightedGroup([1, 2])
447447
cond = _make_weighted_cond(0, group, 1.0, 3.0)
448448
assert cond.__name__ == "weight=1.0 (33%)"
449+
450+
def test_non_zero_index_cond_rolls_dice_if_not_yet_selected(self):
451+
"""When a non-zero index cond is evaluated before index 0, it rolls the dice."""
452+
group = _WeightedGroup([50, 50], seed=42)
453+
cond_1 = _make_weighted_cond(1, group, 50.0, 100.0)
454+
455+
assert group.selected is None
456+
result = cond_1()
457+
assert group.selected is not None # dice was rolled
458+
assert result == (group.selected == 1)

0 commit comments

Comments
 (0)