chore: move left out test config out of functions

himanshusinghs · himanshusinghs · commit d9904608ba15 · 2025-07-17T13:21:19.000+02:00
diff --git a/tests/accuracy/createIndex.test.ts b/tests/accuracy/createIndex.test.ts
@@ -1,30 +1,34 @@
 import { describeAccuracyTests } from "./sdk/describeAccuracyTests.js";
-import { AccuracyTestConfig } from "./sdk/describeAccuracyTests.js";
 
-function callsCreateIndex(prompt: string, indexKeys: Record<string, unknown>): AccuracyTestConfig {
-    return {
-        prompt: prompt,
+describeAccuracyTests([
+    {
+        prompt: "Create an index that covers the following query on 'mflix.movies' namespace - { \"release_year\": 1992 }",
         expectedToolCalls: [
             {
                 toolName: "create-index",
                 parameters: {
                     database: "mflix",
                     collection: "movies",
-                    keys: indexKeys,
+                    keys: {
+                        release_year: 1,
+                    },
                 },
             },
         ],
-    };
-}
-
-describeAccuracyTests([
-    callsCreateIndex(
-        "Create an index that covers the following query on 'mflix.movies' namespace - { \"release_year\": 1992 }",
-        {
-            release_year: 1,
-        }
-    ),
-    callsCreateIndex("Create a text index on title field in 'mflix.movies' namespace", {
-        title: "text",
-    }),
+    },
+    {
+        prompt: "Create a text index on title field in 'mflix.movies' namespace",
+        expectedToolCalls: [
+            {
+                toolName: "create-index",
+                parameters: {
+                    database: "mflix",
+                    collection: "movies",
+                    keys: {
+                        title: "text",
+                    },
+                },
+            },
+        ],
+    },
 ]);
diff --git a/tests/accuracy/updateMany.test.ts b/tests/accuracy/updateMany.test.ts
@@ -1,9 +1,8 @@
 import { describeAccuracyTests } from "./sdk/describeAccuracyTests.js";
-import { AccuracyTestConfig } from "./sdk/describeAccuracyTests.js";
 
-function callsUpdateManyWithEmptyFilters(prompt: string): AccuracyTestConfig {
-    return {
-        prompt: prompt,
+describeAccuracyTests([
+    {
+        prompt: "Update all the documents in 'mflix.movies' namespace with a new field 'new_field' set to 1",
         expectedToolCalls: [
             {
                 toolName: "update-many",
@@ -18,19 +17,16 @@ function callsUpdateManyWithEmptyFilters(prompt: string): AccuracyTestConfig {
                 },
             },
         ],
-    };
-}
-
-function callsUpdateManyWithFilters(prompt: string, filter: Record<string, unknown>): AccuracyTestConfig {
-    return {
-        prompt: prompt,
+    },
+    {
+        prompt: "Update all the documents in 'mflix.movies' namespace, where runtime is less than 100, with a new field 'new_field' set to 1",
         expectedToolCalls: [
             {
                 toolName: "update-many",
                 parameters: {
                     database: "mflix",
                     collection: "movies",
-                    filter,
+                    filter: { runtime: { $lt: 100 } },
                     update: {
                         $set: {
                             new_field: 1,
@@ -39,15 +35,5 @@ function callsUpdateManyWithFilters(prompt: string, filter: Record<string, unkno
                 },
             },
         ],
-    };
-}
-
-describeAccuracyTests([
-    callsUpdateManyWithEmptyFilters(
-        "Update all the documents in 'mflix.movies' namespace with a new field 'new_field' set to 1"
-    ),
-    callsUpdateManyWithFilters(
-        "Update all the documents in 'mflix.movies' namespace, where runtime is less than 100, with a new field 'new_field' set to 1",
-        { runtime: { $lt: 100 } }
-    ),
+    },
 ]);
