def test_choose_die():
bd = BayesDice()
bd.choose_die()
assert bd.die in [4, 6, 8, 12, 20]
def test_roll_die():
bd = BayesDice()
bd.choose_die()
roll = bd.roll_die()
assert roll in range(1, bd.die + 1)
from src.bayesdice import BayesDice
bd = BayesDice()
bd.choose_die()
for i in range(3):
roll = bd.roll_die()
bd.update_priors(roll)