class_docstr: Test WaveVitBlock model
class_docstr_tok: ["Test","WaveVitBlock","model"]
code: def test_Block(self):
      data = torch.randn(1, 16, 64)
      model = Block(dim=64, num_heads=2, mlp_ratio=8, block_type="wave")
      model.float()
      out = model(data, 4, 4)
      self.assertTrue(out.shape, (1, 16, 64))
      model2 = Block(
          dim=64, num_heads=2, mlp_ratio=8, block_type="std_att"
      )
      model2.float()
      out2 = model2(data, 4, 4)
      self.assertTrue(out2.shape, (1, 16, 64))
  
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 4 for calling the TestWaveVitBlock obj's underlying member method code with example usage: obj.test_Block() without return types