add test

Eventual-Inc · Dec 19, 2023 · ddfbfa8 · ddfbfa8
1 parent 7206852
commit ddfbfa8
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/benchmarks/test_concat.py 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)