Dice probabilities

[Thomas Klemm]

(04-30-2024 05:11 PM)Thomas Klemm Wrote:  

• this allows to write unit-tests for some of the functions
  

Here's an example of parametrized tests using pytest:

Code:

import pytest
from dice import calculate_frequencies, coefficients


@pytest.mark.parametrize(
    "raw_die, expected",
    [
        ([0, 1, 2], [1, 1, 1]),
        ([0, 0, 0, 1, 1, 2], [3, 2, 1]),
    ],
)
def test_coefficients(raw_die, expected):
    assert coefficients(raw_die) == expected


@pytest.mark.parametrize(
    "dice, expected",
    [
        ([[1, 1, 1]] * 5, [1, 5, 15, 30, 45, 51, 45, 30, 15, 5, 1]),
        ([[3, 2, 1]] * 5, [243, 810, 1485, 1800, 1590, 1052, 530, 200, 55, 10, 1]),
    ],
)
def test_calculate_frequencies(dice, expected):
    assert calculate_frequencies(dice) == expected

To run the tests, pip install pytest into the virtual environment and execute:

pytest test_dice.py

Code:

================================================================================​============================================ test session starts ===========================================================================​==================================================
platform darwin -- Python 3.12.2, pytest-8.2.0, pluggy-1.5.0
rootdir: /Users/whatever
collected 4 items

test_dice.py ....                                                                                                                                                                                                                                                      [100%]

================================================================================​============================================= 4 passed in 0.01s ===========================================================================​===================================================