code: def test_random_module(nn_module):
      pyro.clear_param_store()
      nn_module = nn_module()
      p = torch.ones(2, 2)
      prior = dist.Bernoulli(p)
      lifted_mod = pyro.random_module("module", nn_module, prior)
      nn_module = lifted_mod()
      for name, parameter in nn_module.named_parameters():
          assert torch.equal(torch.ones(2, 2), parameter.data)
  
imports: import pytest
  import torch
  import torch.nn
  from torch.nn import Parameter
  import pyro
  import pyro.distributions
  import pyro.optim
label_desc: Use image node_id 4 for calling a global function with example usage: test_random_module(nn_module) without return types