def test_make_overloads():
    env_make_hints = {}
    for overloaded_func in typing_extensions.get_overloads(gym.make):
        make_type_hints = typing_extensions.get_type_hints(overloaded_func)
        assert (
            "id" in make_type_hints and "return" in make_type_hints
        ), f"Missing type hints for overloaded `gym.make` id and return, {make_type_hints.get('id', 'unknown env')}"
        if not isinstance(make_type_hints["id"], type):
            for env_id in typing_extensions.get_args(make_type_hints["id"]):
                obs_type, act_type = typing_extensions.get_args(
                    make_type_hints["return"]
                )
                env_make_hints[env_id] = (obs_type, act_type)

    assert (
        env_make_hints.keys() == gym.envs.registry.keys()
    ), f"Missing the make overloads envs: {set(gym.envs.registry.keys()) - set(env_make_hints.keys())}. Additional envs: {set(env_make_hints.keys()) - set(gym.envs.registry.keys())}"
    # todo: add type check on the obs and action type