refactor: bench suites

Author: zsilbi

test/bench/index.ts

@@ -1,16 +1,23 @@
 import { bench, group, run, summary } from "mitata";
+import { serialize } from "../../src/serialize";
 import { benchConfig } from "./config";
-import { isModeEnabled } from "./utils/modes";
 import {
   createBenchObjects,
   getPresetTitle,
   type BenchObjectPreset,
 } from "./utils/objects";
+import { suite } from "./utils/suites";
 import { getVersions } from "./utils/versions";
 
 const versions = await getVersions(benchConfig.versions);
 
-if (isModeEnabled("presets")) {
+versions.push({
+  name: "ohash @ dev",
+  serialize,
+  baseline: true,
+});
+
+suite("presets", () => {
   group("serialize - presets", () => {
     for (const preset of benchConfig.presets) {
       group(getPresetTitle(preset), () => {
@@ -25,9 +32,9 @@ if (isModeEnabled("presets")) {
       });
     }
   });
-}
+});
 
-if (isModeEnabled("combined")) {
+suite("combined", () => {
   group("serialize - combined presets", () => {
     summary(() => {
       for (const version of versions) {
@@ -45,9 +52,9 @@ if (isModeEnabled("combined")) {
       }
     });
   });
-}
+});
 
-if (isModeEnabled("custom")) {
+suite("custom", () => {
   group("serialize - custom", () => {
     group("simple object", () => {
       const object = {
@@ -251,7 +258,7 @@ if (isModeEnabled("custom")) {
       });
     });
   });
-}
+});
 
 await run();
 

test/bench/utils/suites.ts

@@ -0,0 +1,29 @@
+type SuiteName = keyof typeof suites;
+type Suite = (typeof suites)[SuiteName];
+
+const suites = {
+  full: 0b111,
+  presets: 0b100,
+  combined: 0b010,
+  custom: 0b001,
+  undefined: 0b000,
+} as const;
+
+const defaultSuite = suites.presets;
+
+const selectedSuite: Suite =
+  process.argv
+    .slice(2)
+    .map((arg) => {
+      const m = arg.replace("--", "");
+      return m in suites ? suites[m as SuiteName] : suites.undefined;
+    })
+    // eslint-disable-next-line unicorn/no-array-reduce
+    .reduce((acc, val) => (acc | val) as Suite, suites.undefined) ||
+  defaultSuite;
+
+export function suite(mode: SuiteName, fn: () => void) {
+  if (suites[mode] & selectedSuite) {
+    fn();
+  }
+}

test/bench/utils/versions.ts

@@ -7,7 +7,6 @@ import { homedir, tmpdir } from "node:os";
 import { resolve } from "node:path";
 import { pipeline } from "node:stream";
 import { promisify } from "node:util";
-import { serialize } from "../../../src";
 
 /**
  * Supports:
@@ -28,20 +27,12 @@ type Version = {
 
 export async function getVersions(array: VersionString[]): Promise<Version[]> {
   const imports = await Promise.all(array.map((v) => getVersion(v)));
-  const versions: Version[] = array.map((version, i) => ({
+  return array.map((version, i) => ({
     name: `ohash @ ${version.length === 40 ? version.slice(0, 7) : version}`,
     serialize:
       "objectHash" in imports[i] ? imports[i].objectHash : imports[i].serialize,
     baseline: false,
   }));
-
-  versions.push({
-    name: "ohash @ dev",
-    serialize,
-    baseline: true,
-  });
-
-  return versions;
 }
 
 async function getVersion(version: VersionString) {