def testMoveOnDestroyedPiece(self):
board = Board(5, 5)
player = Player()
enemy = Enemy()
enemy.destroyed = True
board.rows[1][1] = player
board.rows[2][2] = enemy
board.move_piece((1, 1), (2, 2))
self.assertEqual(board.rows[2][2], player)
self.assertIsNone(board.rows[1][1])
self.assertEqual(player.hits, MAX_HEALTH)
def testCleanup(self):
board = Board(5, 5)
player = Player()
enemy_one = Enemy()
enemy_one.destroyed = True
enemy_two = Enemy()
bullet_one = Bullet()
bullet_one.destroyed = True
bullet_two = Bullet()
bullet_two.destroyed = True
bullet_three = Bullet()
bonus_one = Bonus()
bonus_one.destroyed = True
bonus_two = Bonus()
board.rows[0][0] = player
board.rows[1][1] = enemy_one
board.rows[2][2] = enemy_two
board.rows[3][3] = bullet_one
board.rows[4][4] = bullet_two
board.rows[1][3] = bullet_three
board.rows[2][0] = bonus_one
board.rows[3][4] = bonus_two
board.cleanup()
self.assertEqual(board.rows[0][0], player)
self.assertEqual(board.rows[2][2], enemy_two)
self.assertEqual(board.rows[1][3], bullet_three)
self.assertEqual(board.rows[3][4], bonus_two)
self.assertIsNone(board.rows[1][1])
self.assertIsNone(board.rows[3][3])
self.assertIsNone(board.rows[4][4])
self.assertIsNone(board.rows[2][0])
