code: def test_data_parallel_setup_module(data_parallel_mock):
      strategy = DataParallelStrategy()
      strategy.parallel_devices = [0, 2, 1]
      module = torch.nn.Linear(2, 2)
      wrapped_module = strategy.setup_module(module)
      assert wrapped_module == data_parallel_mock(
          module=module, device_ids=[0, 2, 1]
      )
  
file_path: pytorch-lightning/tests/tests_fabric/strategies/test_dp.py
imports: from unittest import mock
  from unittest.mock import MagicMock, Mock
  import pytest
  import torch
  from lightning.fabric import Fabric
  from lightning.fabric.strategies import DataParallelStrategy
  from tests_fabric.helpers.runif import RunIf
  from tests_fabric.strategies.test_single_device import _run_test_clip_gradients
label_desc: Use image node_id 3 for calling a global function with example usage: test_data_parallel_setup_module(data_parallel_mock) without return types