game_of_life.py

import numpy as np
import numpy.testing as npt
from itertools import product

b = np.array([
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 1, 0, 0, 0, 0, 1, 0, 1, 0,],
    [0, 0, 1, 0, 0, 0, 0, 1, 0, 0,],
    [0, 0, 0, 1, 0, 0, 0, 0, 1, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 1, 0, 1, 0, 0, 1, 0, 1, 0,],
    [0, 0, 1, 0, 0, 0, 0, 0, 0, 0,],
    [0, 1, 0, 1, 0, 0, 0, 0, 1, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 1, 0, 1, 0, 0, 1, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 1, 0, 1, 0, 0, 0, 0, 1, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
])

b_correct = np.array([
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 1, 0, 0,],
    [0, 0, 1, 0, 0, 0, 0, 1, 1, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 1, 0, 0, 0, 0, 0, 0, 0,],
    [0, 1, 0, 1, 0, 0, 0, 1, 0, 0,],
    [0, 0, 1, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0,],
])

b=(3==(n:=sum(np.roll(b,s,(0,1))for s in product(*2*[(-1,0,1)]))))|(n==4)&b

print(b)
npt.assert_equal(b, b_correct)