def METHOD_NAME():
    from nncf.common.compression import BaseControllerStateNames as CtrlStateNames
    model = get_magnitude_test_model()
    config = get_basic_magnitude_sparsity_config()
    algo_name = config["compression"]["algorithm"]
    config["compression"]["params"] = {"schedule": "multistep"}
    _, compression_ctrl = create_compressed_model_and_algo_for_test(model, config)
    # Test get state
    compression_ctrl.scheduler.load_state({"current_step": 100, "current_epoch": 5})
    state_content = compression_ctrl.get_state()[algo_name]
    assert state_content[CtrlStateNames.SCHEDULER] == {"current_step": 100, "current_epoch": 5}
    # Test load state
    new_state = {
        algo_name: {
            CtrlStateNames.SCHEDULER: {"current_step": 500, "current_epoch": 10},
            CtrlStateNames.LOSS: {},
            CtrlStateNames.COMPRESSION_STAGE: CompressionStage.PARTIALLY_COMPRESSED,
        }
    }
    compression_ctrl.load_state(new_state)
    assert compression_ctrl.scheduler.current_step == 500
    assert compression_ctrl.scheduler.current_epoch == 10
    assert compression_ctrl.get_state() == new_state