From f69b4e9396452fdca72645cb28b199b95a447c4d Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Fri, 31 Mar 2023 04:47:20 +0800
Subject: [PATCH] Add tests for zipformer (#106)

---
 .github/scripts/test-offline-transducer.sh | 39 +++++++++++++---------
 .github/workflows/linux.yaml               |  7 ----
 .github/workflows/macos.yaml               |  6 ----
 .github/workflows/windows-x64.yaml         |  2 ++
 .github/workflows/windows-x86.yaml         |  2 ++
 5 files changed, 28 insertions(+), 28 deletions(-)

diff --git a/.github/scripts/test-offline-transducer.sh b/.github/scripts/test-offline-transducer.sh
index 33b9bec0..6f3fea1d 100755
--- a/.github/scripts/test-offline-transducer.sh
+++ b/.github/scripts/test-offline-transducer.sh
@@ -35,24 +35,33 @@ time $EXE \
   --num-threads=2 \
   $repo/test_wavs/0.wav \
   $repo/test_wavs/1.wav \
-  $repo/test_wavs/2.wav
+  $repo/test_wavs/8k.wav
 
+rm -rf $repo
 
-if command -v sox &> /dev/null; then
-  echo "test 8kHz"
-  sox $repo/test_wavs/0.wav -r 8000 8k.wav
+log "------------------------------------------------------------"
+log "Run Zipformer transducer (English)"
+log "------------------------------------------------------------"
 
-  time $EXE \
-    --tokens=$repo/tokens.txt \
-    --encoder=$repo/encoder-epoch-99-avg-1.onnx \
-    --decoder=$repo/decoder-epoch-99-avg-1.onnx \
-    --joiner=$repo/joiner-epoch-99-avg-1.onnx \
-    --num-threads=2 \
-    $repo/test_wavs/0.wav \
-    $repo/test_wavs/1.wav \
-    $repo/test_wavs/2.wav \
-    8k.wav
-fi
+repo_url=https://huggingface.co/csukuangfj/sherpa-onnx-zipformer-en-2023-03-30
+log "Start testing ${repo_url}"
+repo=$(basename $repo_url)
+log "Download pretrained model and test-data from $repo_url"
+
+GIT_LFS_SKIP_SMUDGE=1 git clone $repo_url
+pushd $repo
+git lfs pull --include "*.onnx"
+popd
+
+time $EXE \
+  --tokens=$repo/tokens.txt \
+  --encoder=$repo/encoder-epoch-99-avg-1.onnx \
+  --decoder=$repo/decoder-epoch-99-avg-1.onnx \
+  --joiner=$repo/joiner-epoch-99-avg-1.onnx \
+  --num-threads=2 \
+  $repo/test_wavs/0.wav \
+  $repo/test_wavs/1.wav \
+  $repo/test_wavs/8k.wav
 
 rm -rf $repo
 
diff --git a/.github/workflows/linux.yaml b/.github/workflows/linux.yaml
index bfb17f8b..bd536e08 100644
--- a/.github/workflows/linux.yaml
+++ b/.github/workflows/linux.yaml
@@ -46,13 +46,6 @@ jobs:
         with:
           fetch-depth: 0
 
-      - name: Install sox
-        shell: bash
-        run: |
-          sudo apt-get update
-          sudo apt-get install -y sox
-          sox -h
-
       - name: Configure CMake
         shell: bash
         run: |
diff --git a/.github/workflows/macos.yaml b/.github/workflows/macos.yaml
index a5d93307..a8a0f1c6 100644
--- a/.github/workflows/macos.yaml
+++ b/.github/workflows/macos.yaml
@@ -43,12 +43,6 @@ jobs:
         with:
           fetch-depth: 0
 
-      - name: Install sox
-        shell: bash
-        run: |
-          brew install sox
-          sox -h
-
       - name: Configure CMake
         shell: bash
         run: |
diff --git a/.github/workflows/windows-x64.yaml b/.github/workflows/windows-x64.yaml
index 39a4b627..5600c2dd 100644
--- a/.github/workflows/windows-x64.yaml
+++ b/.github/workflows/windows-x64.yaml
@@ -7,6 +7,7 @@ on:
     paths:
       - '.github/workflows/windows-x64.yaml'
       - '.github/scripts/test-online-transducer.sh'
+      - '.github/scripts/test-offline-transducer.sh'
       - 'CMakeLists.txt'
       - 'cmake/**'
       - 'sherpa-onnx/csrc/*'
@@ -16,6 +17,7 @@ on:
     paths:
       - '.github/workflows/windows-x64.yaml'
       - '.github/scripts/test-online-transducer.sh'
+      - '.github/scripts/test-offline-transducer.sh'
       - 'CMakeLists.txt'
       - 'cmake/**'
       - 'sherpa-onnx/csrc/*'
diff --git a/.github/workflows/windows-x86.yaml b/.github/workflows/windows-x86.yaml
index eb4022b0..17171b71 100644
--- a/.github/workflows/windows-x86.yaml
+++ b/.github/workflows/windows-x86.yaml
@@ -7,6 +7,7 @@ on:
     paths:
       - '.github/workflows/windows-x86.yaml'
       - '.github/scripts/test-online-transducer.sh'
+      - '.github/scripts/test-offline-transducer.sh'
       - 'CMakeLists.txt'
       - 'cmake/**'
       - 'sherpa-onnx/csrc/*'
@@ -16,6 +17,7 @@ on:
     paths:
       - '.github/workflows/windows-x86.yaml'
       - '.github/scripts/test-online-transducer.sh'
+      - '.github/scripts/test-offline-transducer.sh'
       - 'CMakeLists.txt'
       - 'cmake/**'
       - 'sherpa-onnx/csrc/*'