diff --git a/modin/tests/config/test_envvars.py b/modin/tests/config/test_envvars.py index df1e5ef58e3..debc5514853 100644 --- a/modin/tests/config/test_envvars.py +++ b/modin/tests/config/test_envvars.py @@ -18,6 +18,7 @@ import pytest from packaging import version +import modin.pandas as pd import modin.config as cfg from modin.config.envvars import _check_vars from modin.config.pubsub import _UNSET, ExactStr @@ -111,6 +112,16 @@ def test_doc_module(): assert pandas.read_table.__doc__ in pd.read_table.__doc__ +@pytest.mark.skipif(cfg.Engine.get() != "Ray", reason="Ray specific test") +def test_ray_cluster_resources(): + import ray + + cfg.RayInitCustomResources.put({"special_hardware": 1.0}) + # create a dummy df to initialize Ray engine + _ = pd.DataFrame([1, 2, 3]) + assert ray.cluster_resources()["special_hardware"] == 1.0 + + def test_hdk_envvar(): try: import pyhdk