feat: some slight normalization improvement (#82)

TomerAberbach · Dec 26, 2024 · 6e97f88 · 6e97f88
1 parent c764f17
commit 6e97f88
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/examples/openai-arbitraries.js b/examples/openai-arbitraries.js
@@ -1430,7 +1430,6 @@ export const OpenAI = {
       model: fc.oneof(
         fc.string(),
         fc.constantFrom("text-moderation-latest", "text-moderation-stable"),
-        fc.constant("text-moderation-latest"),
       ),
     },
     {

diff --git a/src/normalize.ts b/src/normalize.ts
@@ -101,15 +101,19 @@ const normalizeDictionaryArbitrary = ({
   dictionaryArbitrary(normalizeArbitrary(key), normalizeArbitrary(value))
 
 const normalizeUnionArbitrary = (arbitrary: UnionArbitrary): Arbitrary => {
-  const variants = new Set(
+  const variants = new Set<Arbitrary>(
     arbitrary.variants.map(normalizeArbitrary).flatMap(variant => {
       // eslint-disable-next-line typescript/switch-exhaustiveness-check
       switch (variant.type) {
         case `enum`:
           // These will be re-collapsed below if necessary.
           return variant.members.map(constantArbitrary)
         case `union`:
-          return variant.variants
+          return variant.variants.flatMap(variant =>
+            variant.type === `enum`
+              ? variant.members.map(constantArbitrary)
+              : [variant],
+          )
         default:
           return [variant]
       }