class TestOutcome(unittest.TestCase):
def setUp(self):
self.outcome1 = Outcome(name="red", odds=17)
self.outcome2 = Outcome("red", 17)
self.outcome3 = Outcome("first-12", 2)
self.outcome4 = Outcome("split-bet", 19)
def test_setup(self):
self.assertEqual(self.outcome1, self.outcome2)
self.assertNotEqual(self.outcome1, self.outcome3)
self.assertEqual(self.outcome1.getName(), self.outcome2.getName())
def test___eq__(self):
self.assertTrue(self.outcome1.__eq__(self.outcome2))
self.assertFalse(self.outcome1.__eq__(self.outcome3))
def test___ne__(self):
self.assertTrue(self.outcome1.__ne__(self.outcome3))
self.assertFalse(self.outcome1.__ne__(self.outcome1))
def test___str__(self):
self.assertEqual(str(self.outcome1), "red (17:1)")
self.assertEqual(self.outcome1.__str__(), "red (17:1)")
def test___repr__(self):
self.assertEqual(self.outcome1.__repr__(), "red (17:1)")
def test_winAmount(self):
self.assertEqual(self.outcome1.winAmount(10), 170)
def test_hash(self):
self.assertEqual(hash(self.outcome1), hash(self.outcome2))
self.assertNotEqual(hash(self.outcome1), hash(self.outcome3))
class WheelConstructionTestCase(unittest.TestCase):
def setUp(self):
# create outcomes
self.outcome_5 = Outcome("00-0-1-2-3", 6)
self.outcome_0 = Outcome("0", 35)
self.outcome_00 = Outcome("00", 35)
# create wheel and bin builder
self.wheel = Wheel()
self.bin_builder = BinBuilder()
def test_add_outcome(self):
# Bin 0 outcomes
self.wheel.add_outcome(0, self.outcome_0)
self.wheel.add_outcome(0, self.outcome_5)
# Bin 00 outcomes
self.wheel.add_outcome(37, self.outcome_00)
self.wheel.add_outcome(37, self.outcome_5)
# assert Bin 0 outcomes
self.assertIn(self.outcome_0, self.wheel.get(0))
self.assertIn(self.outcome_5, self.wheel.get(0))
# assert Bin 00 outcomes
self.assertIn(self.outcome_00, self.wheel.get(37))
self.assertIn(self.outcome_5, self.wheel.get(37))
def test_bin_building(self):
# build bins for the wheel
self.bin_builder.buildBins(self.wheel)
sample_bin1 = [
Outcome("Street 1-2-3", RoulettePayout.StreetBet),
Outcome("Number 1", RoulettePayout.StraightBet),
Outcome("Line 1-2-3-4-5-6", RoulettePayout.LineBet),
Outcome("Low", RoulettePayout.EvenBet),
Outcome("Odd", RoulettePayout.EvenBet),
Outcome("Red", RoulettePayout.EvenBet),
Outcome("Corner 1-2-4-5", RoulettePayout.CornerBet),
Outcome("Dozen 1", RoulettePayout.DozenBet),
Outcome("Column 1", RoulettePayout.ColumnBet),
Outcome("Split 1-2", RoulettePayout.SplitBet),
Outcome("Split 1-4", RoulettePayout.SplitBet),
Outcome("00-0-1-2-3", RoulettePayout.FiveBet)
]
sample_bin2 = [
Outcome("Number 00", RoulettePayout.StraightBet),
Outcome("00-0-1-2-3", RoulettePayout.FiveBet)
]
# check number of outcomes to be equal
self.assertEqual(len(sample_bin1), len(self.wheel.get(1)))
self.assertEqual(len(sample_bin2), len(self.wheel.get(37)))
# check outcomes to be in the bin 1
for outcome in sample_bin1:
self.assertIn(outcome, self.wheel.get(1))
# check outcomes to be in the bin 00
for outcome in sample_bin2:
self.assertIn(outcome, self.wheel.get(37))
def test_outcome_mapping(self):
# add outcomes to wheel
self.wheel.add_outcome(0, self.outcome_0)
self.wheel.add_outcome(37, self.outcome_00)
self.wheel.add_outcome(3, self.outcome_5)
# check that they are mapped
self.assertEqual(self.wheel.getOutcome(self.outcome_0.getName()),
self.outcome_0)
self.assertEqual(self.wheel.getOutcome(self.outcome_00.getName()),
self.outcome_00)
self.assertEqual(self.wheel.getOutcome(self.outcome_5.getName()),
self.outcome_5)
