From 3d4d5ca3df5fcbd697c7d1d58b127ea6f4e301a1 Mon Sep 17 00:00:00 2001
From: Eric Joanis <Eric.Joanis@nrc-cnrc.gc.ca>
Date: Tue, 19 Nov 2024 16:44:41 -0500
Subject: [PATCH] test: add unit testing for this module

---
 aligner/tests/__init__.py |   0
 aligner/tests/test_cli.py | 100 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 100 insertions(+)
 create mode 100644 aligner/tests/__init__.py
 create mode 100644 aligner/tests/test_cli.py

diff --git a/aligner/tests/__init__.py b/aligner/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/aligner/tests/test_cli.py b/aligner/tests/test_cli.py
new file mode 100644
index 0000000..4fdc010
--- /dev/null
+++ b/aligner/tests/test_cli.py
@@ -0,0 +1,100 @@
+"""
+Run wav2vec2aligner unit tests.
+How to run this test suite:
+If you installed wav2vec2aligner:
+    python -m unittest aligner.tests.test_cli
+If you installed everyvoice:
+    python -m unittest everyvoice.model.aligner.wav2vec2aligner.aligner.tests.test_cli
+"""
+
+import os
+import subprocess
+import tempfile
+from pathlib import Path
+from unittest import TestCase
+
+from typer.testing import CliRunner
+
+from ..cli import app
+
+
+class CLITest(TestCase):
+    def setUp(self) -> None:
+        self.runner = CliRunner()
+
+    def test_main_help(self):
+        for help in "-h", "--help":
+            with self.subTest(help=help):
+                result = self.runner.invoke(app, [help])
+                self.assertEqual(result.exit_code, 0)
+                self.assertIn("align", result.stdout)
+                self.assertIn("extract", result.stdout)
+
+    def test_sub_help(self):
+        for cmd in "align", "extract":
+            for help in "-h", "--help":
+                with self.subTest(cmd=cmd, help=help):
+                    result = self.runner.invoke(app, [cmd, help])
+                    self.assertEqual(result.exit_code, 0)
+                    self.assertIn("Usage:", result.stdout)
+                    self.assertIn(cmd, result.stdout)
+
+    def test_align_empty_file(self):
+        with self.subTest("empty file"):
+            result = self.runner.invoke(app, ["align", os.devnull, os.devnull])
+            self.assertNotEqual(result.exit_code, 0)
+            self.assertIn("is empty", result.stdout)
+
+        with self.subTest("file with only empty lines"):
+            with tempfile.TemporaryDirectory() as tmpdir:
+                textfile = os.path.join(tmpdir, "emptylines.txt")
+                with open(textfile, "w", encoding="utf8") as f:
+                    f.write("\n \n   \n")
+                result = self.runner.invoke(app, ["align", textfile, os.devnull])
+                self.assertNotEqual(result.exit_code, 0)
+                self.assertIn("is empty", result.stdout)
+
+    def fetch_ras_test_file(self, filename, outputdir):
+        from urllib.request import Request, urlopen
+
+        repo, path = "https://github.com/ReadAlongs/Studio/", "/test/data/"
+        request = Request(repo + "raw/refs/heads/main" + path + filename)
+        request.add_header("Referer", repo + "blob/main" + path + filename)
+        response = urlopen(request)
+        with open(os.path.join(outputdir, filename), "wb") as f:
+            f.write(response.read())
+
+    def test_align_something(self):
+        with tempfile.TemporaryDirectory() as tmpdir:
+            tmppath = Path(tmpdir)
+            self.fetch_ras_test_file("ej-fra.txt", tmpdir)
+            txt = tmppath / "ej-fra.txt"
+            self.fetch_ras_test_file("ej-fra.m4a", tmpdir)
+            m4a = tmppath / "ej-fra.m4a"
+            wav = tmppath / "ej-fra.wav"
+            # Under most circumstances, align can take a .m4a input file, but not
+            # in CI. Since it's not a hard requirement, so just convert to .wav.
+            subprocess.run(["ffmpeg", "-i", m4a, wav], capture_output=True)
+            # os.system("ls -la " + tmpdir)
+            textgrid = tmppath / "ej-fra-16000.TextGrid"
+            wav_out = tmppath / "ej-fra-16000.wav"
+
+            with self.subTest("ctc-segmenter align"):
+                result = self.runner.invoke(app, ["align", str(txt), str(wav)])
+                if result.exit_code != 0:
+                    os.system("ls -la " + tmpdir)
+                    print(result.stdout)
+                self.assertEqual(result.exit_code, 0)
+                self.assertTrue(textgrid.exists())
+                self.assertTrue(wav_out.exists())
+
+            with self.subTest("ctc-segmenter extract"):
+                result = self.runner.invoke(
+                    app, ["extract", str(textgrid), str(wav_out), str(tmppath / "out")]
+                )
+                if result.exit_code != 0:
+                    print(result.stdout)
+                self.assertEqual(result.exit_code, 0)
+                self.assertTrue((tmppath / "out/metadata.psv").exists())
+                for i in range(7):
+                    self.assertTrue((tmppath / f"out/wavs/segment{i}.wav"))