27 lines
509 B
Python
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__]))
|