def test_wheel_add():
outcome = Outcome("00", 35)
wheel = Wheel(NonRandomNumberGenerator())
wheel.addOutcome(4, outcome)
assert outcome in wheel.get(4)
with pytest.raises(IndexError):
wheel.get(40)
def test_wheel_next():
outcome = Outcome("00", 35)
wheel = Wheel(NonRandomNumberGenerator())
wheel.addOutcome(0, outcome)
# Non random will return first
assert outcome in wheel.next()
outcome = Outcome("00", 35)
wheel = Wheel(NonRandomNumberGenerator())
wheel.addOutcome(1, outcome)
# Non random will return first
assert outcome not in wheel.next()
random_wheel = Wheel()
for i in range(38):
random_wheel.addOutcome(i, outcome)
assert outcome in random_wheel.next()
def test_wheel_get():
outcome = Outcome("00", 35)
wheel = Wheel(NonRandomNumberGenerator())
wheel.addOutcome(4, outcome)
assert outcome in wheel.bins[4]
