chore: import upstream snapshot with attribution

python/ray/tests/test_shuffle.py

@@ -0,0 +1,33 @@
+import sys
+
+import pytest
+
+import ray
+from ray.experimental import shuffle
+
+
+def test_shuffle():
+    try:
+        shuffle.run()
+    finally:
+        ray.shutdown()
+
+
+# https://github.com/ray-project/ray/pull/16408
+@pytest.mark.skipif(sys.platform == "win32", reason="Failing on Windows.")
+def test_shuffle_hang():
+    try:
+        shuffle.run(object_store_memory=1e9, num_partitions=200, partition_size=10e6)
+    finally:
+        ray.shutdown()
+
+
+def test_shuffle_no_streaming():
+    try:
+        shuffle.run(no_streaming=True)
+    finally:
+        ray.shutdown()
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(["-sv", __file__]))