class_docstr: Test WaveVitBlock model
class_docstr_tok: ["Test","WaveVitBlock","model"]
code: def test_Stem(self):
      image = torch.randn(1, 3, 24, 24)
      patch_embed = Stem(3, 20, 10)
      x, _, _ = patch_embed(image)
      self.assertTrue(x.shape, (1, 36, 10))
  
file_path: towhee/tests/unittests/models/wave_vit/test_wave_vit_block.py
imports: import unittest
  import torch
  import numpy
  from towhee.models.wave_vit.wave_vit_block import rand_bbox, Stem, DownSamples, Block, ClassBlock
label_desc: Use image node_id 2 for calling the TestWaveVitBlock obj's underlying member method code with example usage: obj.test_Stem() without return types