diff --git a/tests/test_random_sources.py b/tests/test_random_sources.py index ebf720ee9a11af910bee41aada2727c17488a984..542bae456e6769b0ebba04da4cafb9470f578ef9 100644 --- a/tests/test_random_sources.py +++ b/tests/test_random_sources.py @@ -5,6 +5,7 @@ import sys import argparse from conftest import InputMock from io import StringIO +from itertools import product, chain from diceware import main from diceware.random_sources import ( SystemRandomSource, RealDiceRandomSource, @@ -265,10 +266,8 @@ class TestRealDiceRandomSource(object): src.dice_sides = 4 dist = [0, 0, 0] fake_input( - ["1", "1", "1", "2", "1", "3", "1", "4", - "2", "1", "2", "2", "2", "3", "2", "4", - "3", "1", "3", "2", "3", "3", "3", "4", - "4", "1", "4", "2", "4", "3", "4", "4"]) + list(chain.from_iterable( + product(["1", "2", "3", "4"], repeat=2)))) for x in range(16): picked = src.choice([1, 2, 3]) dist[picked - 1] += 1