98 lines
4.1 KiB
Python
98 lines
4.1 KiB
Python
|
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()
|