def test_grid_translate(): np.random.seed(0) data = np.random.random((10, 15)) layer = Image(data) translate = np.array([15, 15]) layer._translate_grid = translate np.testing.assert_allclose(layer._translate_grid, translate)