chore: import upstream snapshot with attribution

python/ray/data/tests/test_import.py

@@ -0,0 +1,26 @@
+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__]))