code: def test_model(self):
      dummy_img = torch.rand(1, 8, 8, 8).to(self.device)
      out = self.model(dummy_img)
      self.assertEqual(out.shape, (1, 8, 8, 8))
  
imports: import unittest
  import torch
  from towhee.models.poolformer.basic_blocks import basic_blocks
label_desc: Use image node_id 1 for calling the PoolFormerBlockTest obj's underlying member method code with example usage: obj.test_model() without return types