code: def test_reshape_222_to_122():
      src_3d = model_3d_desc(pp_degree=2, tp_degree=2, dp_degree=2)
      tgt_3d = model_3d_desc(pp_degree=1, tp_degree=2, dp_degree=2)
  
      new_3d_map = _do_reshape(src_3d, tgt_3d)
  
      assert new_3d_map[0].get_data(pp_index=0, tp_index=0) == [0, 4]
      assert new_3d_map[0].get_data(pp_index=0, tp_index=1) == [1, 5]
      assert new_3d_map[1].get_data(pp_index=0, tp_index=0) == [2, 6]
      assert new_3d_map[1].get_data(pp_index=0, tp_index=1) == [3, 7]
  
imports: from deepspeed.checkpoint import model_3d_desc
label_desc: Use image node_id 4 for calling a global function with example usage: test_reshape_222_to_122() without return types