diff options
Diffstat (limited to 'benchmarks')
| -rwxr-xr-x | benchmarks/consumer_performance.py | 4 | ||||
| -rwxr-xr-x | benchmarks/producer_performance.py | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/benchmarks/consumer_performance.py b/benchmarks/consumer_performance.py index 5ffd3f5..d7580ce 100755 --- a/benchmarks/consumer_performance.py +++ b/benchmarks/consumer_performance.py @@ -29,7 +29,7 @@ def start_brokers(n): replicas = min(n, 3) print('-> {0} Brokers [{1} partitions / {2} replicas]'.format(n, partitions, replicas)) brokers = [ - KafkaFixture.instance(i, zk.host, zk.port, zk_chroot='', + KafkaFixture.instance(i, zk, zk_chroot='', partitions=partitions, replicas=replicas) for i in range(n) ] @@ -148,7 +148,7 @@ def get_args_parser(): help='Topic for consumer test', default='kafka-python-benchmark-test') parser.add_argument( - '--num-records', type=long, + '--num-records', type=int, help='number of messages to consume', default=1000000) parser.add_argument( diff --git a/benchmarks/producer_performance.py b/benchmarks/producer_performance.py index 0c29cbc..c0de6fd 100755 --- a/benchmarks/producer_performance.py +++ b/benchmarks/producer_performance.py @@ -26,7 +26,7 @@ def start_brokers(n): replicas = min(n, 3) print('-> {0} Brokers [{1} partitions / {2} replicas]'.format(n, partitions, replicas)) brokers = [ - KafkaFixture.instance(i, zk.host, zk.port, zk_chroot='', + KafkaFixture.instance(i, zk, zk_chroot='', partitions=partitions, replicas=replicas) for i in range(n) ] @@ -130,7 +130,7 @@ def get_args_parser(): help='Topic name for test', default='kafka-python-benchmark-test') parser.add_argument( - '--num-records', type=long, + '--num-records', type=int, help='number of messages to produce', default=1000000) parser.add_argument( |