diff --git a/tests/test_random_sources.py b/tests/test_random_sources.py index 97cbcfc35dee012a19eff901857b63285f566b05..d4cbce413d5bbb682d62de4e0296e3c14e3328f0 100644 --- a/tests/test_random_sources.py +++ b/tests/test_random_sources.py @@ -282,3 +282,13 @@ class TestRealDiceRandomSource(object): out, err = capsys.readouterr() assert "roll 2 dice" in out assert picked == 1 + + def test_dice_sides_respected(self, capsys, monkeypatch): + # we use the number of dice sides given by options dict. + self.fake_input_values(["1", "2"], monkeypatch) + src = RealDiceRandomSource(dict(dice_sides=2)) # a coin + picked = src.choice(['a', 'b', 'c', 'd']) + out, err = capsys.readouterr() + # must throw a coin 2 times to pick one out of 4 items + assert "Please roll 2 dice" in out + assert picked == 'b'