Files
2026-07-13 13:17:40 +08:00

27 lines
509 B
Python

import sys
import pytest
from ray._common.test_utils import run_string_as_driver
def test_dynamically_imported():
script = """
import sys
assert "ray" not in sys.modules
import ray
assert "ray" in sys.modules
# `ray.data` shouldn't be imported when `ray` is imported.
assert "ray.data" not in sys.modules
ray.data
# `ray.data` should be cached on import.
assert "ray.data" in sys.modules
"""
run_string_as_driver(script)
if __name__ == "__main__":
sys.exit(pytest.main(["-sv", __file__]))