import Benchmark from "benchmark"; import { z } from "zod/v3"; const doubleSuite = new Benchmark.Suite("z.discriminatedUnion: double"); const manySuite = new Benchmark.Suite("z.discriminatedUnion: many"); const aSchema = z.object({ type: z.literal("a"), }); const objA = { type: "a", }; const bSchema = z.object({ type: z.literal("b"), }); const objB = { type: "b", }; const cSchema = z.object({ type: z.literal("c"), }); const objC = { type: "c", }; const dSchema = z.object({ type: z.literal("d"), }); const double = z.discriminatedUnion("type", [aSchema, bSchema]); const many = z.discriminatedUnion("type", [aSchema, bSchema, cSchema, dSchema]); doubleSuite .add("valid: a", () => { double.parse(objA); }) .add("valid: b", () => { double.parse(objB); }) .add("invalid: null", () => { try { double.parse(null); } catch (_err) { } }) .add("invalid: wrong shape", () => { try { double.parse(objC); } catch (_err) { } }) .on("cycle", (e) => { console.log(`${doubleSuite.name}: ${e.target}`); }); manySuite .add("valid: a", () => { many.parse(objA); }) .add("valid: c", () => { many.parse(objC); }) .add("invalid: null", () => { try { many.parse(null); } catch (_err) { } }) .add("invalid: wrong shape", () => { try { many.parse({ type: "unknown" }); } catch (_err) { } }) .on("cycle", (e) => { console.log(`${manySuite.name}: ${e.target}`); }); export default { suites: [doubleSuite, manySuite], };