def METHOD_NAME():
    return GenericTestEnv(
        action_space=Discrete(4),
        observation_space=Discrete(4),
        reset_func=reset_fn,
        step_func=lambda self, action: (
            action if action < 3 else 0,
            0,
            action >= 3,
            False,
            {"action": action},
        ),
    )