code: def test_fabric_optimizer_state_dict():
      """Test that the FabricOptimizer calls into the strategy to collect the state."""
      optimizer = Mock(spec=torch.optim.Adam)
      strategy = Mock()
      fabric_optimizer = _FabricOptimizer(
          optimizer=optimizer, strategy=strategy
      )
      fabric_optimizer.state_dict()
      strategy.get_optimizer_state.assert_called_with(optimizer)
  
imports: from unittest import mock
  from unittest.mock import Mock, call
  import pytest
  import torch
  from lightning.fabric.fabric import Fabric
  from lightning.fabric.plugins import Precision
  from lightning.fabric.utilities.device_dtype_mixin import _DeviceDtypeModuleMixin
  from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_1
  from lightning.fabric.wrappers import _FabricDataLoader, _FabricModule, _FabricOptimizer, _unwrap_objects, is_wrapped
  from torch.utils.data import BatchSampler, DistributedSampler
  from torch.utils.data.dataloader import DataLoader
  from tests_fabric.helpers.runif import RunIf
label_desc: Use image node_id 12 for calling a global function with example usage: test_fabric_optimizer_state_dict() without return types