code: def test(self, zero_stage, optimizer_constructor):
      config_dict = {
          "train_batch_size": 2,
          "steps_per_print": 1,
          "fp16": {"enabled": True},
          "zero_optimization": {"stage": zero_stage},
      }
      hidden_dim = 10
  
      model = SimpleModel(hidden_dim)
      client_optimizer = optimizer_constructor(
          params=model.parameters()
      )
      model, _, _, _ = deepspeed.initialize(
          config=config_dict, model=model, optimizer=client_optimizer
      )
  
imports: import torch
  import deepspeed.comm
  import deepspeed
  import pytest
  from deepspeed.ops.adam import FusedAdam
  from unit.common import DistributedTest
  from deepspeed.ops.op_builder import CPUAdamBuilder
  from unit.simple_model import SimpleModel, SimpleOptimizer, random_dataloader, SimpleMoEModel, sequence_dataloader
  from unit.util import required_torch_version
label_desc: Use image node_id 1 for calling the TestZeroSupportedClientOptimizer obj's underlying member method code with example usage: obj.test(zero_stage, optimizer_constructor) without return types