Skip to content

Commit d98b219

Browse files
committed
Fix unit test
Signed-off-by: Samuel Monson <[email protected]>
1 parent bb95d1b commit d98b219

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/unit/scheduler/test_strategies.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -335,9 +335,10 @@ def test_string_representation(
335335
self, valid_instances: tuple[ThroughputStrategy, dict]
336336
):
337337
"""Test __str__ method for ThroughputStrategy."""
338-
instance, _ = valid_instances
338+
instance, constructor_args = valid_instances
339+
max_concurrency = constructor_args.get("max_concurrency")
339340
result = str(instance)
340-
assert result == "throughput"
341+
assert result == f"throughput@{max_concurrency or 'unlimited'}"
341342

342343
@pytest.mark.smoke
343344
def test_marshalling(self, valid_instances: tuple[ThroughputStrategy, dict]):

0 commit comments

Comments
 (0)