|
| 1 | +/** |
| 2 | + * Benchmark: dataFramePipeTo — insert DataFrame at a specific argument position in a pipeline. |
| 3 | + * Outputs JSON: {"function": "dataframe_pipe_to", "mean_ms": ..., "iterations": ..., "total_ms": ...} |
| 4 | + */ |
| 5 | +import { DataFrame, dataFramePipeTo } from "../../src/index.ts"; |
| 6 | + |
| 7 | +const SIZE = 100_000; |
| 8 | +const WARMUP = 5; |
| 9 | +const ITERATIONS = 50; |
| 10 | + |
| 11 | +const left = DataFrame.fromColumns({ |
| 12 | + key: Array.from({ length: SIZE }, (_, i) => i % 1000), |
| 13 | + val: Array.from({ length: SIZE }, (_, i) => i * 1.5), |
| 14 | +}); |
| 15 | + |
| 16 | +const right = DataFrame.fromColumns({ |
| 17 | + key: Array.from({ length: 1000 }, (_, i) => i), |
| 18 | + label: Array.from({ length: 1000 }, (_, i) => `item_${i}`), |
| 19 | +}); |
| 20 | + |
| 21 | +// A simple transform: filter df rows where col > threshold |
| 22 | +function filterAbove(threshold: number, df: DataFrame): DataFrame { |
| 23 | + return df.filter((row) => (row["val"] as number) > threshold); |
| 24 | +} |
| 25 | + |
| 26 | +for (let i = 0; i < WARMUP; i++) { |
| 27 | + // dataFramePipeTo inserts `left` at position 1: filterAbove(threshold, left) |
| 28 | + dataFramePipeTo(left, 1, filterAbove, 50_000); |
| 29 | +} |
| 30 | + |
| 31 | +const times: number[] = []; |
| 32 | +for (let i = 0; i < ITERATIONS; i++) { |
| 33 | + const t0 = performance.now(); |
| 34 | + dataFramePipeTo(left, 1, filterAbove, 50_000); |
| 35 | + times.push(performance.now() - t0); |
| 36 | +} |
| 37 | + |
| 38 | +const totalMs = times.reduce((a, b) => a + b, 0); |
| 39 | +const meanMs = totalMs / ITERATIONS; |
| 40 | +console.log( |
| 41 | + JSON.stringify({ |
| 42 | + function: "dataframe_pipe_to", |
| 43 | + mean_ms: meanMs, |
| 44 | + iterations: ITERATIONS, |
| 45 | + total_ms: totalMs, |
| 46 | + }), |
| 47 | +); |
0 commit comments