Mitigate flaky data iter test. (#8244)

- Reduce the number of batches. - Verify labels.
2022-09-14 17:54:14 +08:00
parent bdf265076d
commit 2e63af6117
3 changed files with 13 additions and 3 deletions
--- a/tests/python-gpu/test_gpu_data_iterator.py
+++ b/tests/python-gpu/test_gpu_data_iterator.py
@@ -19,7 +19,7 @@ def test_gpu_single_batch() -> None:
@given(
    strategies.integers(0, 1024),
    strategies.integers(1, 7),
-    strategies.integers(0, 13),
+    strategies.integers(0, 8),
    strategies.booleans(),
 )
@settings(deadline=None, print_blob=True)