loosened test assertion

googleapis · Dec 12, 2024 · e11f701 · e11f701
1 parent 8e3334f
commit e11f701
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/unit/data/_async/test_client.py b/tests/unit/data/_async/test_client.py
@@ -2023,7 +2023,7 @@ async def mock_call(*args, **kwargs):
                     assert isinstance(exc.value, ShardedReadRowsExceptionGroup)
                     # _CONCURRENCY_LIMIT calls will run, and won't be interrupted
                     # calls after the limit will be cancelled due to timeout
-                    assert len(exc.value.exceptions) == num_calls - _CONCURRENCY_LIMIT
+                    assert len(exc.value.exceptions) >= num_calls - _CONCURRENCY_LIMIT
                     assert all(
                         isinstance(e.__cause__, DeadlineExceeded)
                         for e in exc.value.exceptions

diff --git a/tests/unit/data/_sync_autogen/test_client.py b/tests/unit/data/_sync_autogen/test_client.py
@@ -1681,7 +1681,7 @@ def mock_call(*args, **kwargs):
                     with pytest.raises(ShardedReadRowsExceptionGroup) as exc:
                         table.read_rows_sharded(queries, operation_timeout=0.05)
                     assert isinstance(exc.value, ShardedReadRowsExceptionGroup)
-                    assert len(exc.value.exceptions) == num_calls - _CONCURRENCY_LIMIT
+                    assert len(exc.value.exceptions) >= num_calls - _CONCURRENCY_LIMIT
                     assert all(
                         (
                             isinstance(e.__cause__, DeadlineExceeded)