Skip to content

Commit fc23e22

Browse files
committed
fix: fixing pylint Issues
1 parent 7c2bce9 commit fc23e22

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
lines changed

sdk/cosmos/azure-cosmos/azure/cosmos/_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -954,4 +954,4 @@ def format_pk_range_options(query_options: Mapping[str, Any]) -> dict[str, Any]:
954954
pk_range_options["containerRID"] = query_options["containerRID"]
955955
if "excludedLocations" in query_options:
956956
pk_range_options["excludedLocations"] = query_options["excludedLocations"]
957-
return pk_range_options
957+
return pk_range_options

sdk/cosmos/azure-cosmos/tests/test_read_items.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -482,10 +482,14 @@ def test_read_items_concurrency_internals(self):
482482

483483
self.assertEqual(mock_query.call_count, 3)
484484
call_args = mock_query.call_args_list
485-
486-
self.assertEqual(len(call_args[0][0][1]['parameters']), 1000)
487-
self.assertEqual(len(call_args[1][0][1]['parameters']), 1000)
488-
self.assertEqual(len(call_args[2][0][1]['parameters']), 500)
485+
# Extract the number of parameters from each call.
486+
chunk_sizes = [len(call[0][1]['parameters']) for call in call_args]
487+
488+
# Sort the chunk sizes to make the assertion deterministic.
489+
chunk_sizes.sort(reverse=True)
490+
self.assertEqual(chunk_sizes[0], 1000)
491+
self.assertEqual(chunk_sizes[1], 1000)
492+
self.assertEqual(chunk_sizes[2], 500)
489493

490494
def test_read_items_multiple_physical_partitions_and_hook(self):
491495
"""Tests read_items on a container with multiple physical partitions and verifies response_hook."""

0 commit comments

Comments
 (0)