def addPiece(self,owner,type,x,y):
piece = Piece()
piece.setup(owner,type)
if self.array[int(x)][int(y)] == None:
if piece.getOwner()== "AI" and (piece.getType() == 1 or piece.getType() == 0):
self.array[int(x)][int(y)] = piece
self.aiPieces+=1
return True
elif piece.getOwner()== "Player" and (piece.getType() == 1 or piece.getType() == 0):
self.array[int(x)][int(y)] = piece
self.playerPieces+=1
return True
else:
return False
class PieceTest(unittest.TestCase):
def setUp(self):
self.piece = Piece()
# test piece setup
def pieceTest(self):
self.piece.setup("Player", 0)
self.assertEqual("Player", self.piece.getOwner())
self.assertEqual(0, self.piece.getType())
self.piece.setup("AI", 0)
self.assertEqual("AI", self.piece.getOwner())
self.assertEqual(0, self.piece.getType())
self.piece.setup("Player",1)
self.assertEqual("Player", self.piece.getOwner())
self.assertEqual(1, self.piece.getType())
self.piece.setup("AI",1)
self.assertEqual("AI", self.piece.getOwner())
self.assertEqual(1, self.piece.getType())
self.piece.setup(293,"hello")
self.assertEqual(293, self.piece.getOwner())
self.assertEqual("hello", self.piece.getType())
self.piece.setup(1,3)
self.assertEqual(1, self.piece.getOwner())
self.assertEqual(3, self.piece.getType())
def tearDown(self):
self.piece = None
def main(self):
self.setUp()
self.pieceTest()
self.tearDown()
