def test_step_rule_4(self):
initial_state = [[0, 1, 0], [0, 1, 1], [0, 0, 0]]
next_state = np.array([[0, 1, 1], [0, 1, 1], [0, 0, 0]])
game = GameOfLife(initial_state=initial_state)
game.step()
game_state = game.get_state()
self.assertTrue(np.array_equal(game_state, next_state))
def test_run(self):
initial_state = [[1, 1, 1], [1, 1, 1], [1, 1, 1]]
final_state = np.array([[0, 0, 0], [0, 0, 0], [0, 0, 0]])
game = GameOfLife(initial_state=initial_state)
game.run(generations=2)
game_state = game.get_state()
self.assertTrue(np.array_equal(game_state, final_state))