From ddfbfa8198272067a7ef41f8a9606d7cfef05951 Mon Sep 17 00:00:00 2001 From: Sammy Sidhu Date: Tue, 19 Dec 2023 15:07:56 -0800 Subject: [PATCH] add test --- tests/benchmarks/test_concat.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 tests/benchmarks/test_concat.py diff --git a/tests/benchmarks/test_concat.py b/tests/benchmarks/test_concat.py new file mode 100644 index 0000000000..f11b9dd5ef --- /dev/null +++ b/tests/benchmarks/test_concat.py @@ -0,0 +1,16 @@ +from __future__ import annotations + +import uuid + +from daft.series import Series + + +def test_string_concat(benchmark) -> None: + NUM_ROWS = 100_000 + data = Series.from_pylist([str(uuid.uuid4()) for _ in range(NUM_ROWS)]) + to_concat = [data] * 100 + + def bench_concat() -> Series: + return Series.concat(to_concat) + + benchmark(bench_concat)