2 files changed, 88 insertions, 0 deletions
diff --git a/test/unit/__init__.py b/test/unit/__init__.py
new file mode 100644
index 000000000..e69de29bb
--- /dev/null
+++ b/test/unit/__init__.py
diff --git a/test/unit/test_features.py b/test/unit/test_features.py
new file mode 100644
index 000000000..9f9f618b0
--- /dev/null
+++ b/test/unit/test_features.py
@@ -0,0 +1,88 @@
+import unittest
+from scripts.test.shared import WASM_OPT, run_process
+
+
+class FeatureValidationTest(unittest.TestCase):
+  def check_feature(self, module, error, flag):
+    p = run_process(WASM_OPT + ['--mvp-features', '--print'],
+                    input=module, check=False, capture_output=True)
+    self.assertIn(error, p.stderr)
+    self.assertIn("Fatal: error in validating input", p.stderr)
+    self.assertNotEqual(p.returncode, 0)
+    p = run_process(WASM_OPT + ['--mvp-features', flag, '--print'],
+                    input=module, check=False, capture_output=True)
+    self.assertEqual(p.returncode, 0)
+
+  def check_simd(self, module, error):
+    self.check_feature(module, error, '--enable-simd')
+
+  def check_sign_ext(self, module, error):
+    self.check_feature(module, error, '--enable-sign-ext')
+
+  def test_v128_signature(self):
+    module = """
+    (module
+     (func $foo (param $0 v128) (result v128)
+      (local.get $0)
+     )
+    )
+    """
+    self.check_simd(module, "all used types should be allowed")
+
+  def test_v128_global(self):
+    module = """
+    (module
+     (global $foo (mut v128) (v128.const i32x4 0 0 0 0))
+    )
+    """
+    self.check_simd(module, "all used types should be allowed")
+
+  def test_v128_local(self):
+    module = """
+    (module
+     (func $foo
+      (local v128)
+     )
+    )
+    """
+    self.check_simd(module, "all used types should be allowed")
+
+  def test_simd_const(self):
+    module = """
+    (module
+     (func $foo
+      (drop (v128.const i32x4 0 0 0 0))
+     )
+    )
+    """
+    self.check_simd(module, "all used features should be allowed")
+
+  def test_simd_load(self):
+    module = """
+    (module
+     (func $foo
+      (drop (v128.load (i32.const 0)))
+     )
+    )
+    """
+    self.check_simd(module, "SIMD operation (SIMD is disabled)")
+
+  def test_simd_splat(self):
+    module = """
+    (module
+     (func $foo
+      (drop (i32x4.splat (i32.const 0)))
+     )
+    )
+    """
+    self.check_simd(module, "all used features should be allowed")
+
+  def test_sign_ext(self):
+    module = """
+    (module
+     (func $foo
+      (drop (i32.extend8_s (i32.const 7)))
+     )
+    )
+    """
+    self.check_sign_ext(module, "all used features should be allowed")