def test_gen_world(rules): gl = GameOfLife(rules) shape = (50, 50) w = gl.gen_world(shape) assert w.shape == shape shape = (20, 50) w = gl.gen_world(shape) assert w.shape == shape