class Test_Computer(unittest.TestCase): def setUp(self): self.board = Board(6, 7) self.computer = Computer() self.game = Game(self.board,self.computer) def test_make_move_to_win(self): self.board._init_board() self.board.place_value(5,0,2) self.board.place_value(5,1,2) self.board.place_value(5,2,2) self.assertEqual(self.computer.make_move_to_win(self.board),3) self.board._init_board() self.board.place_value(5,0,1) self.board.place_value(5,1,1) self.board.place_value(5,2,1) self.assertEqual(self.computer.make_move_to_win(self.board),3)