Applied suggestions, turned on assertions, added dim order strides to unit tests

Author: Conarnar

extension/wasm/test/CMakeLists.txt

@@ -43,8 +43,13 @@ add_custom_target(
 add_executable(executorch_wasm_tests)
 target_link_libraries(executorch_wasm_tests PUBLIC executorch_wasm)
 target_link_options(
-  executorch_wasm_tests PUBLIC --embed-file "${MODELS_DIR}@/" --post-js
+  executorch_wasm_tests
+  PUBLIC
+  --embed-file
+  "${MODELS_DIR}@/"
+  --post-js
   ${CMAKE_CURRENT_SOURCE_DIR}/unittests.js
+  -sASSERTIONS=2
 )
 set_target_properties(
   executorch_wasm_tests PROPERTIES OUTPUT_NAME "executorch_wasm.test"

extension/wasm/test/unittests.js

@@ -79,6 +79,22 @@ describe("Tensor", () => {
         expect(tensor.scalarType).toEqual(et.ScalarType.Long);
         tensor.delete();
     });
+
+    test("with dim order and strides", () => {
+        const tensor = et.Tensor.fromArray([2, 2], [1, 2, 3, 4], et.ScalarType.Float, [0, 1], [2, 1]);
+        expect(tensor.data).toEqual(new Float32Array([1, 2, 3, 4]));
+        expect(tensor.sizes).toEqual([2, 2]);
+        tensor.delete();
+    });
+
+    test("incorrect dim order", () => {
+        expect(() => et.Tensor.fromArray([2, 2], [1, 2, 3, 4], et.ScalarType.Float, [1])).toThrow();
+        expect(() => et.Tensor.fromArray([2, 2], [1, 2, 3, 4], et.ScalarType.Float, [1, 2])).toThrow();
+    });
+
+    test("incorrect strides", () => {
+        expect(() => et.Tensor.fromArray([2, 2], [1, 2, 3, 4], et.ScalarType.Float, [1, 1], [2, 1])).toThrow();
+    });
 });
 
 describe("Module", () => {