code: def test_grid_distortion_steps(size):
      image = np.random.rand(size, size, 3)
      aug = A.GridDistortion(num_steps=size - 2, p=1)
      data = aug(image=image)
      assert np.array_equal(data["image"].shape, (size, size, 3))
  
imports: import random
  from functools import partial
  import cv2
  import numpy
  import pytest
  import albumentations
  import albumentations.augmentations.functional
  import albumentations.augmentations.geometric.functional
  from albumentations.augmentations.blur.functional import gaussian_blur
  from .utils import get_dual_transforms, get_image_only_transforms, get_transforms
label_desc: Use image node_id 8 for calling a global function with example usage: test_grid_distortion_steps(size) without return types