code: def test_shape_batch(self, device, dtype):
      inp = torch.zeros(2, 2, 4, 4, device=device, dtype=dtype)
      pyr = kornia.geometry.PyrUp()
      assert pyr(inp).shape == (2, 2, 8, 8)
  
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia
  import kornia.testing
  from kornia.testing import BaseTester, assert_close
label_desc: Use image node_id 2 for calling the TestPyrUp obj's underlying member method code with example usage: obj.test_shape_batch(device, dtype) without return types