summaryrefslogtreecommitdiff
path: root/benchmarks/record_batch_compose.py
diff options
context:
space:
mode:
Diffstat (limited to 'benchmarks/record_batch_compose.py')
-rw-r--r--benchmarks/record_batch_compose.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/benchmarks/record_batch_compose.py b/benchmarks/record_batch_compose.py
index 86012df..aca669d 100644
--- a/benchmarks/record_batch_compose.py
+++ b/benchmarks/record_batch_compose.py
@@ -58,7 +58,8 @@ def func(loops, magic):
magic, batch_size=DEFAULT_BATCH_SIZE, compression_type=0)
for _ in range(MESSAGES_PER_BATCH):
key, value, timestamp = next(precomputed_samples)
- size = batch.append(timestamp=timestamp, key=key, value=value)
+ size = batch.append(
+ timestamp=timestamp, key=key, value=value)
assert size
batch.close()
results.append(batch.buffer())
@@ -73,3 +74,4 @@ def func(loops, magic):
runner = perf.Runner()
runner.bench_time_func('batch_append_v0', func, 0)
runner.bench_time_func('batch_append_v1', func, 1)
+runner.bench_time_func('batch_append_v2', func, 2)