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))