code: def test_affine_batch_size_mismatch(self, device, dtype):
      with pytest.raises(RuntimeError):
          angle = torch.rand(1, device=device, dtype=dtype)
          translation = torch.rand(2, 2, device=device, dtype=dtype)
          kornia.geometry.transform.Affine(angle, translation)
  
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia
  import kornia.testing
  from kornia.testing import assert_close
label_desc: Use image node_id 2 for calling the TestAffine2d obj's underlying member method code with example usage: obj.test_affine_batch_size_mismatch(device, dtype) without return types