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)