1 import unittest
   2 
   3 def capsizeCell(aMap, (i,j)):
   4     nMine = 0
   5     for x in range(3):
   6         for y in range(3):
   7             if x==y==1: continue
   8             try:
   9                 if aMap[i-1+x][j-1+y] == '*': 
  10                     nMine+=1
  11             except IndexError:
  12                 continue
  13     return str(nMine)
  14 
  15 def sweepMines(aMap):
  16     result = list()
  17     for i,line in enumerate(aMap):
  18         newline = list()
  19         for j,each in enumerate(line):
  20             if aMap[i][j] == '*':
  21                 neweach = '*'
  22             else:
  23                 neweach = capsizeCell(aMap,(i,j))
  24             newline.append(neweach)
  25         result.append(''.join(newline))
  26     return result
  27 
  28 class MineSweeperTest(unittest.TestCase):
  29     def testExample(self):
  30         inp = [
  31             '*...',
  32             '....',
  33             '.*..',
  34             '....',
  35         ]
  36         expected = [
  37             '*100',
  38             '2210',
  39             '1*10',
  40             '1110',
  41         ]
  42         self.assertEquals(expected, sweepMines(inp))
  43 
  44 if __name__=='__main__':
  45     unittest.main(argv=('','-v'))

   1 def capsizeCell(aMap, (i,j)):
   2     nMine = 0
   3     for x in range(3):
   4         for y in range(3):
   5             cursor_x = i-1+x
   6             cursor_y = j-1+y
   7             try:
   8                 if cursor_x >= 0 and cursor_y >= 0 and aMap[cursor_x][cursor_y] == '*':
   9                     nMine+=1
  10             except IndexError:
  11                 continue
  12     return str(nMine)