Add back tensor map test

huggingface · Oct 10, 2024 · 7a0f77c · 7a0f77c
1 parent 1c43e3f
commit 7a0f77c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/utils/tensor.test.js b/tests/utils/tensor.test.js
@@ -129,6 +129,16 @@ describe("Tensor operations", () => {
     });
   });
 
+  describe("map", () => {
+    it("should double", async () => {
+      const original = new Tensor("float32", [1, 2, 3, 4, 5, 6], [2, 3]);
+      const target = new Tensor("float32", [2, 4, 6, 8, 10, 12], [2, 3]);
+
+      const doubled = original.map((x) => x * 2);
+      compare(doubled, target, 1e-3);
+    });
+  });
+
   describe("mean", () => {
     it("should calculate mean", async () => {
       const t1 = new Tensor("float32", [1, 2, 3, 4, 5, 6], [2, 3, 1]);