Generate a square RGB test image of the given side length.

def _image_of_size(image_size):
  """Generate a square RGB test image of the given side length."""
  return np.random.uniform(0, 256, [image_size, image_size, 3]).astype(np.uint8)