class_docstr: Test WaveVitUtils model
code: def test_DWT2D(self):
      xx = torch.randn(1, 12, 12, 12)
      i_dwt = IDWT2D(wave="haar")
      i_dwt.float()
      out = i_dwt(xx)
      self.assertTrue(out.shape, (1, 3, 24, 24))
  
imports: import unittest
  import torch
  from towhee.models.wave_vit.wave_vit_utils import IDWT2D, DWT2D
label_desc: Use image node_id 2 for calling the TestWaveVitUtils obj's underlying member method code with example usage: obj.test_DWT2D() without return types