squiggle/packages/squiggle-lang/scripts/bench-map.mjs

#!/usr/bin/env node
import { SqProject } from "@quri/squiggle-lang";

const measure = (cb, times = 1) => {
  const t1 = new Date();

  for (let i = 1; i <= times; i++) {
    cb();
  }
  const t2 = new Date();
  return (t2 - t1) / 1000;
};

const maxP = 7;

for (let p = 0; p <= maxP; p++) {
  const size = Math.pow(10, p);
  const project = SqProject.create();
  project.setSource("list", `l = List.upTo(1,${size})`);
  project.run("list");
  project.setSource("map", "l |> map({|x| x})");
  project.setContinues("map", ["list"]);
  const time = measure(() => {
    project.run("map");
  });
  console.log(`1e${p}`, "\t", time);
}