def test_weight_locusts(self):
maxDiff = 2048
class BaseLocust(Locust):
pass
class L1(BaseLocust):
weight = 101
class L2(BaseLocust):
weight = 99
class L3(BaseLocust):
weight = 100
runner = Environment(locust_classes=[L1, L2, L3]).create_local_runner()
self.assert_locust_class_distribution({
L1: 10,
L2: 9,
L3: 10
}, runner.weight_locusts(29))
self.assert_locust_class_distribution({
L1: 10,
L2: 10,
L3: 10
}, runner.weight_locusts(30))
self.assert_locust_class_distribution({
L1: 11,
L2: 10,
L3: 10
}, runner.weight_locusts(31))
def test_weight_locusts_fewer_amount_than_locust_classes(self):
class BaseLocust(Locust):
pass
class L1(BaseLocust):
weight = 101
class L2(BaseLocust):
weight = 99
class L3(BaseLocust):
weight = 100
runner = Environment(locust_classes=[L1, L2, L3]).create_local_runner()
self.assertEqual(1, len(runner.weight_locusts(1)))
self.assert_locust_class_distribution({L1: 1},
runner.weight_locusts(1))
