test

ray-project · Mar 3, 2023 · 6b7130b · 6b7130b
1 parent d0c6a1f
commit 6b7130b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/ray/data/tests/test_util.py b/python/ray/data/tests/test_util.py
@@ -82,6 +82,9 @@ def test_list_splits():
     assert _split_list(list(range(5)), 7) == [[0], [1], [2], [3], [4], [], []]
     assert _split_list(list(range(5)), 2) == [[0, 1, 2], [3, 4]]
     assert _split_list(list(range(6)), 2) == [[0, 1, 2], [3, 4, 5]]
+    assert _split_list(list(range(5)), 1) == [[0, 1, 2, 3, 4]]
+    assert _split_list(["foo", 1, [0], None], 2) == [["foo", 1], [[0], None]]
+    assert _split_list(["foo", 1, [0], None], 3) == [["foo", 1], [[0]], [None]]
 
 
 if __name__ == "__main__":