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()