def test_weight_locusts(self):
maxDiff = 2048
class BaseLocust(Locust):
class task_set(TaskSet):
pass
class L1(BaseLocust):
weight = 101
class L2(BaseLocust):
weight = 99
class L3(BaseLocust):
weight = 100
runner = LocustRunner([L1, L2, L3], mocked_options())
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(self):
maxDiff = 2048
class BaseLocust(Locust):
pass
class L1(BaseLocust):
weight = 101
class L2(BaseLocust):
weight = 99
class L3(BaseLocust):
weight = 100
runner = LocustRunner(Environment(options=mocked_options()),
locust_classes=[L1, L2, L3])
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):
class task_set(TaskSet): pass
class L1(BaseLocust):
weight = 101
class L2(BaseLocust):
weight = 99
class L3(BaseLocust):
weight = 100
runner = LocustRunner([L1, L2, L3], mocked_options())
self.assertEqual(1, len(runner.weight_locusts(1)))
self.assert_locust_class_distribution({L1:1}, runner.weight_locusts(1))
