Brunobkr/llama.cpp_AlgMor24_github

1// @ts-ignore TS61332import { expect, test } from "vitest";3 4import * as z from "zod/v3";5import { util } from "../helpers/util.js";6 7test("nativeEnum test with consts", () => {8  const Fruits: { Apple: "apple"; Banana: "banana" } = {9    Apple: "apple",10    Banana: "banana",11  };12  const fruitEnum = z.nativeEnum(Fruits);13  type fruitEnum = z.infer<typeof fruitEnum>;14  fruitEnum.parse("apple");15  fruitEnum.parse("banana");16  fruitEnum.parse(Fruits.Apple);17  fruitEnum.parse(Fruits.Banana);18  util.assertEqual<fruitEnum, "apple" | "banana">(true);19});20 21test("nativeEnum test with real enum", () => {22  enum Fruits {23    Apple = "apple",24    Banana = "banana",25  }26  // @ts-ignore27  const fruitEnum = z.nativeEnum(Fruits);28  type fruitEnum = z.infer<typeof fruitEnum>;29  fruitEnum.parse("apple");30  fruitEnum.parse("banana");31  fruitEnum.parse(Fruits.Apple);32  fruitEnum.parse(Fruits.Banana);33  util.assertIs<fruitEnum extends Fruits ? true : false>(true);34});35 36test("nativeEnum test with const with numeric keys", () => {37  const FruitValues = {38    Apple: 10,39    Banana: 20,40    // @ts-ignore41  } as const;42  const fruitEnum = z.nativeEnum(FruitValues);43  type fruitEnum = z.infer<typeof fruitEnum>;44  fruitEnum.parse(10);45  fruitEnum.parse(20);46  fruitEnum.parse(FruitValues.Apple);47  fruitEnum.parse(FruitValues.Banana);48  util.assertEqual<fruitEnum, 10 | 20>(true);49});50 51test("from enum", () => {52  enum Fruits {53    Cantaloupe = 0,54    Apple = "apple",55    Banana = "banana",56  }57 58  const FruitEnum = z.nativeEnum(Fruits as any);59  type _FruitEnum = z.infer<typeof FruitEnum>;60  FruitEnum.parse(Fruits.Cantaloupe);61  FruitEnum.parse(Fruits.Apple);62  FruitEnum.parse("apple");63  FruitEnum.parse(0);64  expect(() => FruitEnum.parse(1)).toThrow();65  expect(() => FruitEnum.parse("Apple")).toThrow();66  expect(() => FruitEnum.parse("Cantaloupe")).toThrow();67});68 69test("from const", () => {70  const Greek = {71    Alpha: "a",72    Beta: "b",73    Gamma: 3,74    // @ts-ignore75  } as const;76 77  const GreekEnum = z.nativeEnum(Greek);78  type _GreekEnum = z.infer<typeof GreekEnum>;79  GreekEnum.parse("a");80  GreekEnum.parse("b");81  GreekEnum.parse(3);82  expect(() => GreekEnum.parse("v")).toThrow();83  expect(() => GreekEnum.parse("Alpha")).toThrow();84  expect(() => GreekEnum.parse(2)).toThrow();85 86  expect(GreekEnum.enum.Alpha).toEqual("a");87});88