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]