locust 2.47.1.dev41__tar.gz → 2.47.1.dev43__tar.gz

{locust-2.47.1.dev41 → locust-2.47.1.dev43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: locust
-Version: 2.47.1.dev41
+Version: 2.47.1.dev43
 Summary: Developer-friendly load testing framework
 Project-URL: Homepage, https://locust.cloud/
 Project-URL: Repository, https://github.com/locustio/locust

{locust-2.47.1.dev41 → locust-2.47.1.dev43}/locust/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '2.47.1.dev41'
-__version_tuple__ = version_tuple = (2, 47, 1, 'dev41')
+__version__ = version = '2.47.1.dev43'
+__version_tuple__ = version_tuple = (2, 47, 1, 'dev43')
 
 __commit_id__ = commit_id = None

{locust-2.47.1.dev41 → locust-2.47.1.dev43}/locust/user/inspectuser.py

@@ -4,6 +4,7 @@ import inspect
 from collections import defaultdict
 from json import dumps
 
+from ..dispatch import _kl_generator
 from .task import TaskSet
 from .users import User
 
@@ -32,11 +33,15 @@ def _calc_distribution(user_classes, num_users):
     num_users = num_users or (total_weight if not fixed_count else 1)
     weighted_count = num_users - fixed_count
     weighted_count = max(0, weighted_count)
-    user_classes_count = {}
-
-    for u in user_classes:
-        count = u.fixed_count or (u.weight / total_weight) * weighted_count
-        user_classes_count[u.__name__] = round(count)
+    user_classes_count = {u.__name__: u.fixed_count for u in user_classes}
+
+    # Pick weighted users the way the dispatcher does, so the counts match a real run.
+    weighted_users = _kl_generator((u.__name__, u.weight) for u in user_classes if not u.fixed_count)
+    for _ in range(int(weighted_count)):
+        name = next(weighted_users)
+        if name is None:
+            break
+        user_classes_count[name] += 1
 
     return user_classes_count