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