def test_case1(self): data = (".......", ".......", ".......", "...R...", "...BB..", "..BRB..", ".RRBR..") winner = get_winner(data, 3) self.assertEqual(winner, None)
def test_case3(self): data = ("R...", "BR..", "BR..", "BR..") winner = get_winner(data, 4) self.assertEqual(winner, Winners.RED)
def test_case4(self): data = ("B..", "RB.", "RB.") winner = get_winner(data, 3) self.assertEqual(winner, Winners.BLUE)
def test_case2(self): data = ("......", "......", ".R...R", ".R..BB", ".R.RBR", "RB.BBB") winner = get_winner(data, 4) self.assertEqual(winner, Winners.BOTH)