def setUp(self): unittest.TestCase.setUp(self) self.monty = MonteCarloIterative() self.monty.color = 1
class Test(unittest.TestCase): def setUp(self): unittest.TestCase.setUp(self) self.monty = MonteCarloIterative() self.monty.color = 1 def test_move(self): board=Board(6,7) move = self.monty.play(board) self.assertTrue(move in range(7)) def test_add_score(self): column = 0 self.monty.add_score(column, 0) self.assertEqual(-1, self.monty.score(column)) def test_play_one_round(self): board = Board(6, 7) played_move = self.monty.play(board) self.assertTrue(played_move in range(7)) def test_add_score_for_other_color(self): self.monty.color = 0 column = 0 self.monty.add_score(column, self.monty.color) self.assertEqual(1, self.monty.score(column)) def test_add_score2(self): column = 0 self.monty.add_score(column, self.monty.color) self.assertEqual(1, self.monty.score(column)) def test_add_score3(self): column = 0 self.monty.add_score(column, None) self.assertEqual(0, self.monty.score(column))