code: def test_class(self, device, dtype):
      img = torch.zeros((5, 1, 5, 5))
      labels = torch.randint(0, 10, size=(5, 1))
      aug = K.AugmentationSequential(
          K.RandomCrop((3, 3), pad_if_needed=True),
          data_keys=["input", "class"],
      )
  
      _, out_labels = aug(img, labels)
      assert labels is out_labels
  
imports: from functools import partial
  from unittest.mock import patch
  import pytest
  import torch
  import kornia
  import kornia.augmentation
  from kornia.augmentation.container.base import ParamItem
  from kornia.constants import BorderType
  from kornia.geometry.bbox import bbox_to_mask
  from kornia.testing import assert_close
label_desc: Use image node_id 13 for calling the TestAugmentationSequential obj's underlying member method code with example usage: obj.test_class(device, dtype) without return types