class_docstr: Test WaveVitBlock model
class_docstr_tok: ["Test","WaveVitBlock","model"]
code: def test_rand_bbox(self):
      beta = 1.0
      pooling_scale = 8
      lam = np.random.beta(beta, beta)
      x = torch.randn(1, 24, 24, 3)
      bbx1, bby1, bbx2, bby2 = rand_bbox(
          x.size(), lam, scale=pooling_scale
      )
      self.assertTrue(bbx1.dtype, np.int64)
      self.assertTrue(bby1.dtype, np.int64)
      self.assertTrue(bbx2.dtype, np.int64)
      self.assertTrue(bby2.dtype, np.int64)
  
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 1 for calling the TestWaveVitBlock obj's underlying member method code with example usage: obj.test_rand_bbox() without return types