/
moore_neighborhood.py
38 lines (34 loc) · 1.42 KB
/
moore_neighborhood.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#Input: Three arguments. A grid as a tuple of tuples with integers (1/0),
## a row number and column number for a cell as integers.
#Output: How many neighbouring cells have chips as an integer.
def count_neighbours(grid, row, col):
count = 0
neighbors = ((-1, -1), (-1, 0), (-1, 1), (0, -1),
(0, 1), (1, -1), (1, 0), (1, 1))
for item in neighbors:
n_row = row + item[0]
n_col = col + item[1]
if 0 <= n_row < len(grid) and 0 <= n_col < len(grid[n_row]):
if grid[n_row][n_col] == 1:
count += 1
return count
#From Checkio:
if __name__ == '__main__':
#These "asserts" using only for self-checking and not necessary for auto-testing
assert count_neighbours(
((1, 0, 0, 1, 0),
(0, 1, 0, 0, 0),
(0, 0, 1, 0, 1),
(1, 0, 0, 0, 0),
(0, 0, 1, 0, 0),), 1, 2) == 3, "1st example"
assert count_neighbours(((1, 0, 0, 1, 0),
(0, 1, 0, 0, 0),
(0, 0, 1, 0, 1),
(1, 0, 0, 0, 0),
(0, 0, 1, 0, 0),), 0, 0) == 1, "2nd example"
assert count_neighbours(((1, 1, 1),
(1, 1, 1),
(1, 1, 1),), 0, 2) == 3, "Dense corner"
assert count_neighbours(((0, 0, 0),
(0, 1, 0),
(0, 0, 0),), 1, 1) == 0, "Single"