ray-project--ray
wehub-resource-sync
19 行
396 B
Python
19 行
396 B
Python
import sys
|
|
|
|
import pytest
|
|
|
|
from ray._private.collections_utils import split
|
|
|
|
|
|
def test_split():
|
|
ints = list(range(0, 5))
|
|
|
|
assert list(split(ints, 5)) == [ints]
|
|
assert list(split(ints, 10)) == [ints]
|
|
assert list(split(ints, 1)) == [[e] for e in ints]
|
|
assert list(split(ints, 2)) == [[0, 1], [2, 3], [4]]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-v", __file__]))
|