code: def test(self, zero_stage, use_cpu_offload):
      if (
          use_cpu_offload
          and not deepspeed.ops.__compatible_ops__[CPUAdamBuilder.NAME]
      ):
          pytest.skip("cpu-adam is not compatible")
  
      config_dict = {
          "train_batch_size": 4,
          "steps_per_print": 1,
          "fp16": {
              "enabled": True,
          },
          "zero_optimization": {
              "stage": zero_stage,
              "cpu_offload": use_cpu_offload,
          },
          "zero_allow_untested_optimizer": False,
          "zero_force_ds_cpu_optimizer": False,
      }
      hidden_dim = 10
  
      model = SimpleModel(hidden_dim)
      optimizer = SimpleOptimizer(model.parameters())
      with pytest.raises(AssertionError):
          model, optim, _, _ = deepspeed.initialize(
              config=config_dict,
              model=model,
              optimizer=optimizer,
              model_parameters=model.parameters(),
          )
  
imports: import torch
  import deepspeed.comm
  import deepspeed
  import pytest
  from deepspeed.ops.adam import FusedAdam
  from unit.common import DistributedTest
  from unit.simple_model import SimpleModel, SimpleOptimizer, random_dataloader, SimpleMoEModel, sequence_dataloader
  from deepspeed.runtime.utils import required_torch_version
  from deepspeed.accelerator import get_accelerator
  from deepspeed.ops.op_builder import CPUAdamBuilder
label_desc: Use image node_id 1 for calling the TestZeroAllowUntestedOptimizer obj's underlying member method code with example usage: obj.test(zero_stage, use_cpu_offload) without return types