code: def test_copy_param_shape_check(self):
      a = nn.Linear(2, 5)
      b = nn.Linear(1, 5)
  
      with self.assertRaises(RuntimeError):
          # Different shape
          copy_param.copy_param(a, b)
  
      with self.assertRaises(RuntimeError):
          # Different shape
          copy_param.copy_param(b, a)
  
file_path: pfrl/tests/utils_tests/test_copy_param.py
imports: import unittest
  import numpy
  import torch
  import torch.nn
  from pfrl.testing import torch_assert_allclose
  from pfrl.utils import copy_param
label_desc: Use image node_id 3 for calling the TestCopyParam obj's underlying member method code with example usage: obj.test_copy_param_shape_check() without return types