chore: import upstream snapshot with attribution

python/ray/data/tests/doctest_pytest_plugin.py

@@ -0,0 +1,25 @@
+"""This file is injected for Ray Data doctest targets."""
+import pytest
+
+import ray
+
+
+@pytest.fixture(autouse=True, scope="module")
+def shutdown_ray():
+    ray.shutdown()
+    yield
+
+
+@pytest.fixture(autouse=True)
+def preserve_block_order():
+    ray.data.context.DataContext.get_current().execution_options.preserve_order = True
+    yield
+
+
+@pytest.fixture(autouse=True)
+def disable_start_message():
+    context = ray.data.context.DataContext.get_current()
+    original_value = context.print_on_execution_start
+    context.print_on_execution_start = False
+    yield
+    context.print_on_execution_start = original_value