Skip to content

Commit

Permalink
Add a test
Browse files Browse the repository at this point in the history
Signed-off-by: Igoshev, Iaroslav <[email protected]>
  • Loading branch information
YarShev committed Apr 22, 2024
1 parent c763161 commit 36f5d7b
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions modin/tests/config/test_envvars.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 36f5d7b

Please sign in to comment.