@m4trix/evals 0.31.0 → 0.32.0

package/dist/cli-simple.js

@@ -320,6 +320,23 @@ function isRunConfigLike(value) {
 function isTestCaseLike(value) {
   return hasMethod(value, "getName") && hasMethod(value, "getTags") && hasMethod(value, "getInput");
 }
+function collectTestCasesFromExportValues(exports) {
+  const out = [];
+  for (const value of exports) {
+    if (isTestCaseLike(value)) {
+      out.push(value);
+      continue;
+    }
+    if (Array.isArray(value)) {
+      for (const item of value) {
+        if (isTestCaseLike(item)) {
+          out.push(item);
+        }
+      }
+    }
+  }
+  return out;
+}
 async function walkDirectory(rootDir, excludeDirectories) {
   const out = [];
   async function walk(currentDir) {
@@ -428,7 +445,7 @@ async function collectTestCasesFromFiles(config) {
   const found = await Promise.all(
     matched.map(async (absolutePath) => {
       const exports = await loadModuleExports(absolutePath);
-      const testCases = exports.filter(isTestCaseLike);
+      const testCases = collectTestCasesFromExportValues(exports);
       const relPath = relative(config.rootDir, absolutePath);
       return testCases.map((testCase) => ({
         id: toId("test-case", relPath, testCase.getName()),