class_docstr: Test WaveVitBlock model
class_docstr_tok: ["Test","WaveVitBlock","model"]
code: def test_DownSamples(self):
      xx = torch.randn(5, 10, 3, 3)
      patch_embed = DownSamples(10, 5)
      x, _, _ = patch_embed(xx)
      self.assertTrue(x.shape, (5, 4, 5))
  
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 3 for calling the TestWaveVitBlock obj's underlying member method code with example usage: obj.test_DownSamples() without return types