def testGetOutcomeByName(self):
# arrange
w = Wheel()
o = Outcome('red', 1)
# act
w.addOutcome(5, o)
# assert
self.assertEqual(o, w.getOutcomeByName('red'))
def test_randomness(self):
# arrange
pseudoRandom = random.Random()
pseudoRandom.seed(1)
randomIndices = [pseudoRandom.randint(0, 37) for _ in range(10)]
outcomes = [Outcome(i, 1) for i in range(10)]
zipped = zip(randomIndices, outcomes)
w = Wheel()
for z in zipped:
w.addOutcome(z[0], z[1])
w.rng.seed(1)
# act+assert
for _ in zipped:
self.assertEqual(w.next(), Bin({outcomes[i]}))
class TestOutcome(unittest.TestCase):
def setUp(self):
self.wheel = Wheel()
self.outcome0 = Outcome("Number 0", 36)
self.outcome00 = Outcome("Number 00", 36)
self.bin0 = Bin([self.outcome0])
self.bin00 = Bin([self.outcome00])
def test_addOutcomesToWheel(self):
self.wheel.addOutcome("0", self.outcome0)
self.wheel.addOutcome("00", self.outcome00)
def test_binIncludedInWheel(self):
self.assertIn(self.bin0, self.wheel.bins)
self.assertIn(self.bin00, self.wheel.bins)
def testAddOutcome(self):
w = Wheel()
o = Outcome('red', 1)
w.addOutcome(5, o)
self.assertEqual(Bin({o}), w.getBinByIndex(5))
