Support openWakeWord

tests/test_microwakeword.py

@@ -0,0 +1,43 @@
+"""Tests for microWakeWord."""
+
+import wave
+from pathlib import Path
+
+from linux_voice_assistant.microwakeword import MicroWakeWord, MicroWakeWordFeatures
+from linux_voice_assistant.util import is_arm
+
+_TESTS_DIR = Path(__file__).parent
+_REPO_DIR = _TESTS_DIR.parent
+_MICRO_DIR = _REPO_DIR / "wakewords"
+
+if is_arm():
+    _LIB_DIR = _REPO_DIR / "lib" / "linux_arm64"
+else:
+    _LIB_DIR = _REPO_DIR / "lib" / "linux_amd64"
+
+
+libtensorflowlite_c_path = _LIB_DIR / "libtensorflowlite_c.so"
+
+
+def test_features() -> None:
+    features = MicroWakeWordFeatures(
+        libtensorflowlite_c_path=libtensorflowlite_c_path,
+    )
+    ww = MicroWakeWord.from_config(
+        config_path=_MICRO_DIR / "okay_nabu.json",
+        libtensorflowlite_c_path=libtensorflowlite_c_path,
+    )
+
+    detected = False
+    with wave.open(str(_TESTS_DIR / "ok_nabu.wav"), "rb") as wav_file:
+        assert wav_file.getframerate() == 16000
+        assert wav_file.getsampwidth() == 2
+        assert wav_file.getnchannels() == 1
+
+        for micro_input in features.process_streaming(
+            wav_file.readframes(wav_file.getnframes())
+        ):
+            if ww.process_streaming(micro_input):
+                detected = True
+
+    assert detected