continued unittesting

tests/test_minimax.py

@@ -0,0 +1,71 @@
+import unittest
+from minimax import Minimax
+
+class TestMinimax(unittest.TestCase):
+    def setUp(self):
+        self.minimax = Minimax()  # Create an instance of your Minimax class
+
+    def test_minimax(self):
+        current_field = [["X", "|", " ", "|", " "],
+                         ["—", "+", "—", "+", "—"],
+                         [" ", "|", "X", "|", " "],
+                         ["—", "+", "—", "+", "—"],
+                         ["O", "|", "O", "|", " "]]  # Set the current field state
+        player1_char = 'X'
+        player2_char = 'O'
+
+        score, x, y = self.minimax.minimax(current_field, True, "X", "O")  # Call the minimax method
+        expected_score = 1  # Adjust the expected score based on the game rules and the field state
+        expected_x = 4  # Adjust the expected x-coordinate of the best move
+        expected_y = 4  # Adjust the expected y-coordinate of the best move
+
+        self.assertEqual(score, expected_score)  # Check if the score matches the expected value
+        self.assertEqual(x, expected_x)  # Check if the x-coordinate of the best move matches the expected value
+        self.assertEqual(y, expected_y)  # Check if the y-coordinate of the best move matches the expected value
+
+    def test_check_win(self) -> None:
+        player1_char = 'X'
+        player2_char = 'O'
+
+        # Testing player 1 win condition
+        field = [["X", "|", " ", "|", " "],
+                 ["—", "+", "—", "+", "—"],
+                 ["X", "|", " ", "|", " "],
+                 ["—", "+", "—", "+", "—"],
+                 ["X", "|", " ", "|", " "]]
+
+        result = self.minimax.check_win(field, player1_char, player2_char)
+        self.assertEqual(result, (1, 0))
+
+        # Testing player 2 win condition
+        field = [["O", "|", " ", "|", " "],
+                 ["—", "+", "—", "+", "—"],
+                 ["O", "|", " ", "|", " "],
+                 ["—", "+", "—", "+", "—"],
+                 ["O", "|", " ", "|", " "]]
+
+        result = self.minimax.check_win(field, player1_char, player2_char)
+        self.assertEqual(result, (0, 1))
+
+        # Testing draw condition
+        field = [["X", "|", "O", "|", "X"],
+                 ["—", "+", "—", "+", "—"],
+                 ["X", "|", "O", "|", "O"],
+                 ["—", "+", "—", "+", "—"],
+                 ["O", "|", "X", "|", "X"]]
+
+        result = self.minimax.check_win(field, player1_char, player2_char)
+        self.assertEqual(result, (1, 1))
+
+        # Testing no win condition
+        field = [[" ", "|", " ", "|", " "],
+                 ["—", "+", "—", "+", "—"],
+                 [" ", "|", " ", "|", " "],
+                 ["—", "+", "—", "+", "—"],
+                 [" ", "|", " ", "|", " "]]
+
+        result = self.minimax.check_win(field, player1_char, player2_char)
+        self.assertEqual(result, (0, 0))
+
+if __name__ == '__main__':
+    unittest.main()