wehub-resource-sync
34 lines
638 B
Python
34 lines
638 B
Python
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__]))
|