chore: import upstream snapshot with attribution

python/ray/tests/test_distributed_sort.py

@@ -0,0 +1,26 @@
+import sys
+
+import pytest
+
+from ray.experimental.raysort import main
+
+
+@pytest.mark.skipif(sys.platform == "win32", reason="Failing on Windows")
+def test_distributed_sort():
+    args = main.get_args(
+        [
+            "--total_data_size=1_000_000_000",
+            "--num_mappers=4",
+            "--num_reducers=4",
+            "--num_mappers_per_round=2",
+            "--ray_address=",
+            "--skip_input",
+            "--skip_output",
+        ]
+    )
+    main.main(args)
+
+
+if __name__ == "__main__":
+
+    sys.exit(pytest.main(["-sv", __file__]))