code: def test_forward(self):
      pse_head = PSEHead(
          in_channels=[10], hidden_dim=128, out_channel=7
      )
      results = pse_head(self.feature)
      self.assertEqual(results.shape, (2, 7, 40, 50))
  
imports: from unittest import TestCase
  import torch
  from mmocr.models.textdet.heads import PSEHead
label_desc: Use image node_id 3 for calling the TestPSEHead obj's underlying member method code with example usage: obj.test_forward() without return types