class RaiseBotTest(unittest.TestCase): def setUp(self): self.bot = RaiseBot([], [], []) self.bot.add_brain() self.bot.bet = self.bet self.raised = 0 def bet(self, amount): self.raised = amount def test_raise_bot(self): self.bot.brain.do_turn("bot", 100) self.assertGreater(self.raised, 0)