continued unittesting

tests/test_model.py

@@ -0,0 +1,97 @@
+import unittest
+from model import Model
+import os.path
+
+
+class TestModel(unittest.TestCase):
+    def setUp(self) -> None:
+        # self.model = model.Model()
+        self.field = [[" ", "|", " ", "|", " "],
+                      ["—", "+", "—", "+", "—"],
+                      [" ", "|", " ", "|", " "],
+                      ["—", "+", "—", "+", "—"],
+                      [" ", "|", " ", "|", " "]]
+        self.field2 = [["X", "|", " ", "|", " "],
+                       ["—", "+", "—", "+", "—"],
+                       [" ", "|", " ", "|", " "],
+                       ["—", "+", "—", "+", "—"],
+                       [" ", "|", " ", "|", " "]]
+        self.model = Model()  # Create an instance of your Model class
+
+    def tearDown(self):
+        if os.path.isfile('./savestate.py'):
+            os.remove("savestate.py")  # Remove the savestate file if it exists
+
+    def test_loadsavestate(self):
+        f = open("savestate.py", "w")  # Create a savestate file
+        f.write('save = [["X", "|", " ", "|", " "],'
+                '["—", "+", "—", "+", "—"],'
+                '[" ", "|", " ", "|", " "],'
+                '["—", "+", "—", "+", "—"],'
+                '[" ", "|", " ", "|", " "]]')
+        f.close()
+
+        self.model.loadsavestate()  # Call the loadsavestate method
+        expected_field = [["X", "|", " ", "|", " "],
+                          ["—", "+", "—", "+", "—"],
+                          [" ", "|", " ", "|", " "],
+                          ["—", "+", "—", "+", "—"],
+                          [" ", "|", " ", "|", " "]]
+        self.assertEqual(self.model.field, expected_field)  # Check if the field matches the expected state
+
+    def test_checksavestate(self):
+        self.assertFalse(self.model.checksavestate())  # Check if savestate does not exist
+
+        f = open("savestate.py", "w")  # Create a savestate file
+        f.write('save = [["X", "|", " ", "|", " "],'
+                '["—", "+", "—", "+", "—"],'
+                '[" ", "|", " ", "|", " "],'
+                '["—", "+", "—", "+", "—"],'
+                '[" ", "|", " ", "|", " "]]')
+        f.close()
+
+        self.assertTrue(self.model.checksavestate())  # Check if savestate exists
+
+    def test_savetosavestate(self):
+        self.model.field = [["X", "|", " ", "|", " "],
+                            ["—", "+", "—", "+", "—"],
+                            [" ", "|", " ", "|", " "],
+                            ["—", "+", "—", "+", "—"],
+                            [" ", "|", " ", "|", " "]]  # Set the field to a specific state
+
+        self.model.savetosavestate()  # Call the savetosavestate method
+
+        f = open("savestate.py", "r")  # Read the content of the savestate file
+        content = f.read()
+        f.close()
+
+        expected_content = "save = [['X', '|', ' ', '|', ' ']," \
+                           " ['—', '+', '—', '+', '—'], [' ', '|', ' ', '|', ' ']," \
+                           " ['—', '+', '—', '+', '—'], [' ', '|', ' ', '|', ' ']]"
+        self.assertEqual(content, expected_content)  # Check if the content matches the expected value
+
+    def test_deletesavestate(self):
+        f = open("savestate.py", "w")  # Create a savestate file
+        f.close()
+
+        self.model.deletesavestate()  # Call the deletesavestate method
+
+        self.assertFalse(os.path.isfile('./savestate.py'))  # Check if the savestate file does not exist
+
+    def test_check_move(self):
+        model = Model()  # Create an instance of your Model class
+        self.assertEqual(model.check_move(0, 0, True), True)
+
+    def test_do_move(self):
+        model = Model()  # Create an instance of your Model class
+        model.do_move(0, 0, True, "X", "O")  # Call the do_move method
+        expected_field = [["X", "|", " ", "|", " "],
+                          ["—", "+", "—", "+", "—"],
+                          [" ", "|", " ", "|", " "],
+                          ["—", "+", "—", "+", "—"],
+                          [" ", "|", " ", "|", " "]]
+        self.assertEqual(model.field, expected_field)
+
+
+if __name__ == '__main__':
+    unittest.main()