all tests

Author: jgw96

tests/main.test.ts

@@ -1,42 +1,66 @@
 import { expect, test } from 'vitest';
 
-test('speech-to-text', async () => {
+test('text-to-speech', async () => {
     return new Promise(async (resolve) => {
-        const { transcribeAudioFile } = await import("../src/index");
+        const { textToSpeech } = await import("../src/index");
+
+        const audio = await textToSpeech("Hello, World!");
+        expect(audio).toBeDefined();
 
-        const input = document.createElement("button");
+        resolve(true);
+    });
+});
 
-        input.onclick = async (e) => {
-            const response = await fetch("https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/ted_60_16k.wav");
-            const blob = await response.blob();
+test('speech-to-text', async () => {
+    return new Promise(async (resolve) => {
+        const { transcribeAudioFile } = await import("../src/index");
 
-            const text = await transcribeAudioFile(blob);
-            console.log(text);
-            expect(text).toBeDefined();
+        const response = await fetch("https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/ted_60_16k.wav");
+        const blob = await response.blob();
 
-            resolve(true);
-        };
+        const text = await transcribeAudioFile(blob);
+        expect(text).toBeDefined();
 
-        document.body.appendChild(input);
+        resolve(true);
     });
 });
 
 test('ocr', async () => {
     return new Promise(async (resolve) => {
         const { ocr } = await import("../src/index");
 
-        const input = document.createElement("input");
-        input.type = "file";
-        input.accept = "image/*";
+        const response = await fetch("https://picsum.photos/200/300");
+        const blob = await response.blob();
+
+        const text = await ocr(blob);
+
+        expect(text).toBeDefined();
+        resolve(true);
+    });
+});
+
+test('image-classification', async () => {
+    return new Promise(async (resolve) => {
+        const { classifyImage } = await import("../src/index");
+
+        const response = await fetch("https://picsum.photos/200/300");
+        const blob = await response.blob();
 
-        input.onchange = async (e) => {
-            const file = (e.target as any).files[0];
-            const text = await ocr(URL.createObjectURL(file) as any);
+        const text = await classifyImage(blob);
+
+        expect(text).toBeDefined();
+        resolve(true);
+    });
+});
+
+test('summarize', async () => {
+    return new Promise(async (resolve) => {
+        const { summarize } = await import("../src/index");
 
-            expect(text).toBeDefined();
-            resolve(true);
-        };
+        const text = "the red fox is a small fox that lives in the forest. it has a red coat and a bushy tail. the red fox is a carnivore, which means it eats meat. it hunts small animals like rabbits";
+        const summary = await summarize(text);
 
-        document.body.appendChild(input);
+        expect(summary).toBeDefined();
+        resolve(true);
     });
 });

vitest.workspace.ts

@@ -9,7 +9,7 @@ export default defineWorkspace([
       testTimeout: 2000000,
       browser: {
         enabled: true,
-        name: 'edge-canary',
+        name: 'edge',
         provider: 'preview',
       },
     },