def testMaskBoard(self):
testObject = Board()
model = ModelBattleship()
model.placeShip(testObject, 5, [0, 0], 'v')
model.placeShip(testObject, 4, [0, 1], 'v')
model.placeShip(testObject, 3, [0, 2], 'v')
model.placeShip(testObject, 2, [0, 3], 'v')
model.placeShip(testObject, 1, [0, 4], 'v')
model.placeShot(testObject, [0, 0])
maskedBoard = model.maskBoard(testObject)
actualValue = testObject
expectedValue = maskedBoard
assert (actualValue != expectedValue)
flag = True
while flag:
controller = ControllerBattleship()
controller.playerBoard = Board()
controller.enemyBoard = Board()
view = ViewBattleship()
model = ModelBattleship()
userInput = view.displayMenu()
if userInput[0] == '1':
controller.makeAIBoard()
view.displayBoard(controller.playerBoard)
controller.inputShips()
if userInput[1] == '1':
while 1:
view.display("Enemy Board: ")
view.displayBoard(model.maskBoard(controller.enemyBoard))
controller.getShot()
if model.checkWin(controller.enemyBoard):
view.display("Congrats you won!!")
break
else:
model.randAI(controller.playerBoard)
view.display("Your Board: ")
view.displayBoard(controller.playerBoard)
if model.checkWin(controller.playerBoard):
view.display("Oh no you lost!!")
break
view.displayBoard(controller.enemyBoard)
elif userInput[1] == '2':
while 1:
view.display("Enemy Board: ")