1. TestCaseBoard.py ¶
~cpp
import unittest
from Board import *
class BoardTestCase (unittest.TestCase):
def setUp (self):
self.dolboard = Board ()
self.dolboard.Empty ()
def tearDown (self):
self.dolboard = None
def testBoardEmpty (self):
self.dolboard.Empty ()
for i in range (0,19):
for j in range (0,19):
self.assertEquals (self.dolboard.GetDol(i,j) , DolEmpty)
def testIsExistDolInPosition (self):
self.dolboard.SetDol (1,1,DolWhite)
self.assertEquals (self.dolboard.IsExistDolInPosition(1,1),1)
self.assertEquals (self.dolboard.IsExistDolInPosition(3,6),0)
def testDol (self):
self.dolboard.SetDol (1,1,DolWhite)
self.assertEquals (self.dolboard.GetDol(1,1), DolWhite)
self.dolboard.SetDol (2,2,DolBlack)
self.assertEquals (self.dolboard.GetDol(2,2), DolBlack)
self.assertEquals (self.dolboard.GetDol(18,18), DolEmpty)
self.assertEquals (self.dolboard.GetDol(6,6), DolEmpty)
self.dolboard.SetDol (18,18,DolBlack)
self.assertEquals (self.dolboard.GetDol(18,18), DolBlack)
def testPrintEmptyBoard (self):
self.dolboard.Empty ()
self.assertEquals (self.dolboard.Print(), '''+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
+++++++++++++++++++
''')
suite = unittest.makeSuite (BoardTestCase, "test")
runner = unittest.TextTestRunner ()
runner.run (suite)
2. TestCaseOmok.py ¶
~cpp
import unittest
from Omok import *
class OmokTestCase (unittest.TestCase):
def setUp (self):
self.omok = Omok ()
self.omok.Start ()
def tearDown (self):
self.omok = None
def testIsExistDolInPosition (self):
self.omok.PutDol (2,2)
self.assertEquals (self.omok.IsExistDolInPosition (2,2),1)
self.omok.PutDol (0,0)
self.assertEquals (self.omok.IsExistDolInPosition (0,0),1)
self.assertEquals (self.omok.IsExistDolInPosition (16,16),0)
def testChangeCurrentDol (self):
currentDol = self.omok.GetCurrentDol ()
self.omok.ChangeCurrentDol ()
self.assertNotEqual (currentDol, self.omok.GetCurrentDol ())
def testFirstDolIsBlackInStart (self):
self.omok.Start ()
self.assertEquals (self.omok.GetCurrentDol (), DolBlack)
def testPutDol (self):
self.omok.PutDol (3,4)
self.assertEquals (self.omok.GetDol (3,4), DolBlack)
def testCurrentDol (self):
self.omok.PutDol (3,4)
self.assertEquals (self.omok.GetCurrentDol(), DolBlack)
self.omok.PutDol (9,6)
self.assertEquals (self.omok.GetCurrentDol(), DolBlack)
def testCheckFiveLeftTopToRightBottom (self):
self.omok.PutDol (0,0) # black
self.omok.PutDol (1,1) # black
self.omok.PutDol (2,2) # black
self.omok.PutDol (3,3) # black
self.omok.PutDol (4,4) # black
self.assertEquals (self.omok.CheckFiveLeftTopToRightBottom (4,4),1)
self.omok.Start ()
self.omok.PutDol (18,18) # black
self.omok.PutDol (17,17) # black
self.omok.PutDol (16,16) # black
self.omok.PutDol (15,15) # black
self.omok.PutDol (14,14) # black
self.assertEquals (self.omok.CheckFiveLeftTopToRightBottom (14,14),1)
self.omok.Start ()
self.omok.SetCurrentDol (DolWhite)
self.omok.PutDol (5,4) # white
self.omok.PutDol (6,5) # white
self.omok.PutDol (7,6) # white
self.omok.PutDol (8,7) # white
self.omok.PutDol (9,8) # white
self.assertEquals (self.omok.CheckFiveLeftTopToRightBottom (9,8),1)
def testCheckFiveLeftBottomToRightTop (self):
self.omok.PutDol (0,5) # black
self.omok.PutDol (1,4) # black
self.omok.PutDol (2,3) # black
self.omok.PutDol (3,2) # black
self.omok.PutDol (4,1) # black
self.assertEquals (self.omok.CheckFiveLeftBottomToRightTop (4,1),1)
self.omok.Start ()
self.omok.PutDol (18,14) # black
self.omok.PutDol (17,15) # black
self.omok.PutDol (16,16) # black
self.omok.PutDol (15,17) # black
self.omok.PutDol (14,18) # black
self.assertEquals (self.omok.CheckFiveLeftBottomToRightTop (14,18),1)
self.omok.Start ()
self.omok.SetCurrentDol (DolWhite)
self.omok.PutDol (5,8) # white
self.omok.PutDol (6,7) # white
self.omok.PutDol (7,6) # white
self.omok.PutDol (8,5) # white
self.omok.PutDol (9,4) # white
self.assertEquals (self.omok.CheckFiveLeftBottomToRightTop (9,4),1)
def testCheckFiveHorizon (self):
self.omok.PutDol (0,0)
self.omok.PutDol (1,0)
self.omok.PutDol (2,0)
self.omok.PutDol (3,0)
self.omok.PutDol (4,0)
self.assertEquals (self.omok.CheckFiveHorizon (4,0), 1)
self.omok.Start ()
self.omok.PutDol (18,0)
self.omok.PutDol (17,0)
self.omok.PutDol (16,0)
self.omok.PutDol (15,0)
self.omok.PutDol (14,0)
self.assertEquals (self.omok.CheckFiveHorizon (14,0), 1)
self.omok.Start ()
self.omok.PutDol (5,3)
self.omok.PutDol (6,3)
self.omok.PutDol (7,3)
self.omok.PutDol (8,3)
self.omok.PutDol (9,3)
self.assertEquals (self.omok.CheckFiveHorizon (9,3), 1)
def testCheckFiveVertical (self):
self.omok.PutDol (0,0)
self.omok.PutDol (0,1)
self.omok.PutDol (0,2)
self.omok.PutDol (0,3)
self.omok.PutDol (0,4)
self.assertEquals (self.omok.CheckFiveVertical (0,4), 1)
self.omok.Start ()
self.omok.PutDol (0,18)
self.omok.PutDol (0,17)
self.omok.PutDol (0,16)
self.omok.PutDol (0,15)
self.omok.PutDol (0,14)
self.assertEquals (self.omok.CheckFiveVertical (0,14), 1)
self.omok.Start ()
self.omok.PutDol (3,5)
self.omok.PutDol (3,6)
self.omok.PutDol (3,7)
self.omok.PutDol (3,8)
self.omok.PutDol (3,9)
self.assertEquals (self.omok.CheckFiveVertical (3,9), 1)
suite = unittest.makeSuite (OmokTestCase, "test")
runner = unittest.TextTestRunner ()
runner.run (suite)
3. Board.py ¶
~cpp
import sys
DolEmpty = 9
DolBlack = 0
DolWhite = 1
class Board:
def __init__(self):
self.boardArray = {}
def Empty(self):
for i in range (0,19):
for j in range (0,19):
self.boardArray[(i,j)] = 9
def GetDol (self,x,y):
return self.boardArray[(y,x)]
def SetDol (self,x,y,dol):
self.boardArray[(y,x)] = dol
def IsExistDolInPosition (self,x,y):
if self.GetDol (x,y) != DolEmpty:
return 1
else:
return 0
def Print (self):
PrintString = ''
for i in range (0,19):
for j in range (0,19):
dol = "%s" % self.boardArray[(i,j)]
if dol == '9':
dol = '+'
PrintString += dol
PrintString += "\n"
return PrintString
4. Omok.py ¶
~cpp
from Board import *
import string
class Omok:
def __init__(self):
self.currentDol = DolEmpty
self.board = Board ()
self.currentWinner = DolEmpty
pass
def Start (self):
self.currentDol = DolBlack
self.board.Empty ()
pass
def GetCurrentDol (self):
return self.currentDol
def SetCurrentDol (self, dol):
self.currentDol = dol
def ChangeCurrentDol (self):
if self.currentDol == DolBlack:
self.SetCurrentDol (DolWhite)
elif self.currentDol == DolWhite:
self.SetCurrentDol (DolBlack)
def CheckFiveHorizon (self,x,y):
for i in range (max(0,x-4),min (19,x+5)):
CountSameDol = 0
for j in range (i,min(19,i+5)):
if self.board.GetDol (j,y) == self.currentDol:
CountSameDol += 1
if CountSameDol == 5:
return 1
return 0
def CheckFiveLeftTopToRightBottom (self,x,y):
# Left-top -> Right-Bottom
# \
# \
# >
for i in range (-4,5):
CountSameDol = 0
for j in range (i,i+5):
if x + j < 0 or x + j > 18:
continue
elif y + j < 0 or y + j > 18:
continue
else:
if self.board.GetDol (x+j,y+j) == self.currentDol:
CountSameDol += 1
if CountSameDol == 5:
return 1
return 0
def CheckFiveLeftBottomToRightTop (self,x,y):
# Left-Bottom -> Right-Top
# >
# /
# /
for i in range (-4,5):
CountSameDol = 0
for j in range (i,i+5):
if x + j < 0 or x + j > 18:
continue
elif y - j < 0 or y - j > 18:
continue
else:
if self.board.GetDol (x+j,y-j) == self.currentDol:
CountSameDol += 1
if CountSameDol == 5:
return 1
return 0
def CheckFiveVertical (self,x,y):
for i in range (max(0,y-4),min (19,y+5)):
CountSameDol = 0
for j in range (i,min(19,i+5)):
if self.board.GetDol (x,j) == self.currentDol:
CountSameDol += 1
if CountSameDol == 5:
return 1
return 0
def PutDol (self,x,y):
self.board.SetDol (x,y,self.currentDol)
def GetDol (self,x,y):
return self.board.GetDol (x,y)
def CheckForFive (self,x,y):
if self.CheckFiveHorizon (x,y) == 1:
self.currentWinner = self.currentDol
if self.CheckFiveVertical (x,y) == 1:
self.currentWinner = self.currentDol
if self.CheckFiveLeftTopToRightBottom (x,y) == 1:
self.currentWinner = self.currentDol
def CurrentWinner (self):
return self.currentWinner
def QuestionDolPosition (self):
Position = raw_input ("Input Dol Position x,y :")
Position = string.split (Position, ",")
Position[0] = (int)(Position[0])
Position[1] = (int)(Position[1])
return Position
def IsExistDolInPosition (self,x,y):
return self.board.IsExistDolInPosition (x,y)
def main (self):
self.Start ()
while self.CurrentWinner() == DolEmpty:
print self.board.Print ()
Position = self.QuestionDolPosition ()
x,y = Position
if not self.IsExistDolInPosition (x,y):
self.PutDol (x,y)
self.CheckForFive (x,y)
self.ChangeCurrentDol ()
print "Current Winner : %s" % self.CurrentWinner ()
if __name__ == '__main__':
omok = Omok ()
omok.main ()
