School Commit Init
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
import unittest
|
||||
|
||||
from src.Board.board import Board
|
||||
from src.AI.ai import AI
|
||||
|
||||
class AI_Tester(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self.ai=AI()
|
||||
self.board=Board()
|
||||
return super().setUp()
|
||||
def tearDown(self) -> None:
|
||||
self.ai=None
|
||||
return super().tearDown()
|
||||
def test_get_move_random(self):
|
||||
self.board.reset()
|
||||
for i in range(6):
|
||||
self.board.place(0,i%2+1)
|
||||
self.board.place(1,i%2+1)
|
||||
self.board.place(2,(i+1)%2+1)
|
||||
self.board.place(3,(i+1)%2+1)
|
||||
self.board.place(5,i%2+1)
|
||||
self.board.place(6,i%2+1)
|
||||
self.assertEqual(self.ai.get_move(self.board),4)
|
||||
def test_get_move_win(self):
|
||||
self.board.reset()
|
||||
self.board.place(0,2)
|
||||
self.board.place(1,2)
|
||||
self.board.place(2,2)
|
||||
self.assertEqual(self.ai.get_move(self.board),3)
|
||||
def test_get_move_block(self):
|
||||
self.board.reset()
|
||||
self.board.place(0,1)
|
||||
self.board.place(1,1)
|
||||
self.board.place(2,1)
|
||||
self.assertEqual(self.ai.get_move(self.board),3)
|
||||
+49
@@ -0,0 +1,49 @@
|
||||
import unittest
|
||||
|
||||
from src.Board.board import Board
|
||||
|
||||
class Board_Tester(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self.board=Board()
|
||||
return super().setUp()
|
||||
def tearDown(self) -> None:
|
||||
self.board=None
|
||||
return super().tearDown()
|
||||
|
||||
def test_place(self):
|
||||
self.board.reset()
|
||||
for i in range(6):
|
||||
self.board.place(0,1)
|
||||
self.assertEqual(self.board.check_cell(i,0),1)
|
||||
self.assertRaises(Exception,self.board.place,0,1)
|
||||
def test_is_full(self):
|
||||
self.board.reset()
|
||||
for i in range(7):
|
||||
for j in range(6):
|
||||
self.assertFalse(self.board.is_full())
|
||||
self.board.place(i,1)
|
||||
self.assertTrue(self.board.is_full())
|
||||
def test_reset(self):
|
||||
self.board.reset()
|
||||
self.board.place(0,1)
|
||||
self.board.reset()
|
||||
self.assertEqual(self.board.check_cell(0,0),None)
|
||||
def test_get_board(self):
|
||||
self.board.reset()
|
||||
for i in range(7):
|
||||
for j in range(6):
|
||||
self.board.place(i,1)
|
||||
board_state=self.board.get_board()
|
||||
for i in range(7):
|
||||
for j in range(6):
|
||||
self.assertEqual(board_state[i][j],1)
|
||||
def test_check_cell(self):
|
||||
self.board.reset()
|
||||
for i in range(7):
|
||||
for j in range(6):
|
||||
self.board.place(i,1)
|
||||
for i in range(7):
|
||||
for j in range(6):
|
||||
self.assertEqual(self.board.check_cell(j,i),1)
|
||||
|
||||
|
||||
+52
@@ -0,0 +1,52 @@
|
||||
from src.Game.game import Game
|
||||
from src.Board.board import Board
|
||||
from src.AI.ai import AI
|
||||
import unittest
|
||||
|
||||
class Game_Tester(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self.board=Board()
|
||||
self.ai=AI()
|
||||
self.game=Game(self.board,self.ai)
|
||||
return super().setUp()
|
||||
|
||||
def tearDown(self) -> None:
|
||||
self.game=None
|
||||
return super().tearDown()
|
||||
|
||||
def test_play(self):
|
||||
self.game.reset_game()
|
||||
self.board.reset()
|
||||
self.board.place(0,1)
|
||||
self.board.place(1,1)
|
||||
self.game.play(3)
|
||||
self.assertEqual(self.board.check_cell(0,0),1)
|
||||
self.assertEqual(self.board.check_cell(0,1),1)
|
||||
self.assertEqual(self.board.check_cell(0,2),1)
|
||||
self.assertEqual(self.board.check_cell(0,3),2)
|
||||
|
||||
def test_get_score(self):
|
||||
self.game.reset_game()
|
||||
self.assertListEqual(self.game.get_score(),[0,0])
|
||||
self.board.reset()
|
||||
self.board.place(0,1)
|
||||
self.board.place(1,1)
|
||||
self.board.place(2,1)
|
||||
self.game.play(4)
|
||||
self.assertListEqual(self.game.get_score(),[1,0])
|
||||
|
||||
def test_reset_board(self):
|
||||
self.game.reset_board()
|
||||
count=0
|
||||
for i in range(7):
|
||||
if self.board.check_cell(0,i)==None:
|
||||
count+=1
|
||||
self.assertGreaterEqual(count,6)
|
||||
def test_reset_game(self):
|
||||
self.game.reset_game()
|
||||
count=0
|
||||
for i in range(7):
|
||||
if self.board.check_cell(0,i)==None:
|
||||
count+=1
|
||||
self.assertGreaterEqual(count,6)
|
||||
self.assertListEqual(self.game.get_score(),[0,0])
|
||||
Reference in New Issue
Block a user