def setUp(self):
self.bot = CallRaiseBot([], [], [])
self.bot.add_brain()
self.bot.bet = self.bet
self.bot.call = self.call
self.raises = 0
self.calls = 0
class CallRaiseBotTest(unittest.TestCase):
def setUp(self):
self.bot = CallRaiseBot([], [], [])
self.bot.add_brain()
self.bot.bet = self.bet
self.bot.call = self.call
self.raises = 0
self.calls = 0
def bet(self, amount):
self.raises += 1
def call(self, amount):
self.calls += 1
def test_call_raise_bot(self):
for _ in range(50):
self.bot.brain.do_turn("bot", 100)
self.assertGreater(self.raises, 10)
self.assertGreater(self.calls, 10)
