Update package smoke test

.github/workflows/build-wheels-m1.yml

@@ -36,7 +36,7 @@ jobs:
             cache-key: macos-m1-ffmpeg
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
-            smoke-test-script: test/smoke_test/smoke_test.py
+            smoke-test-script: test/smoke_test/smoke_test_no_ffmpeg.py
             package-name: torchaudio
     name: ${{ matrix.repository }}
     uses: pytorch/test-infra/.github/workflows/build_wheels_macos.yml@main

.github/workflows/build_wheels_linux.yml

@@ -36,7 +36,7 @@ jobs:
             cache-key: linux-ffmpeg
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
-            smoke-test-script: test/smoke_test/smoke_test.py
+            smoke-test-script: test/smoke_test/smoke_test_no_ffmpeg.py
             package-name: torchaudio
     name: ${{ matrix.repository }}
     uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main

.github/workflows/build_wheels_macos.yml

@@ -36,7 +36,7 @@ jobs:
             cache-key: macos-ffmpeg
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
-            smoke-test-script: test/smoke_test/smoke_test.py
+            smoke-test-script: test/smoke_test/smoke_test_no_ffmpeg.py
             package-name: torchaudio
     name: ${{ matrix.repository }}
     uses: pytorch/test-infra/.github/workflows/build_wheels_macos.yml@main

.github/workflows/build_wheels_windows.yml

@@ -34,7 +34,7 @@ jobs:
             cache-key: windows-ffmpeg
             wheel-build-params: "--plat-name win_amd64"
             post-script: ""
-            smoke-test-script: test/smoke_test/smoke_test.py
+            smoke-test-script: test/smoke_test/smoke_test_no_ffmpeg.py
             package-name: torchaudio
     name: ${{ matrix.repository }}
     uses: pytorch/test-infra/.github/workflows/build_wheels_windows.yml@main

test/smoke_test/smoke_test.py

@@ -1,3 +1,4 @@
+#!/usr/bin/env python3
 """Run smoke tests"""
 import argparse
 import logging
@@ -19,26 +20,44 @@ def ffmpeg_test():
     from torchaudio.io import StreamReader  # noqa: F401
 
 
-def main() -> None:
-    options = _parse_args()
+def _run_smoke_test(check_ffmpeg):
+    base_smoke_test()
+
+    if not check_ffmpeg:
+        print("Skipping ffmpeg test.")
+    else:
+        ffmpeg_test()
+
+    print("Smoke test passed.")
+
+
+def main(args=None) -> None:
+    options = _parse_args(args)
 
     if options.debug:
         logging.basicConfig(format="%(levelname)s: %(message)s", level=logging.DEBUG)
 
-    base_smoke_test()
-    if options.ffmpeg:
-        ffmpeg_test()
-    print("Smoke test passed.")
+    _chdir()
+    _run_smoke_test(options.ffmpeg)
 
 
-def _parse_args():
+def _parse_args(args):
     parser = argparse.ArgumentParser()
 
     # Warning: Please note this option should not be widely used, only use it when absolutely necessary
     parser.add_argument("--no-ffmpeg", dest="ffmpeg", action="store_false")
     parser.add_argument("--debug", action="store_true", help="Enable debug logging.")
 
-    return parser.parse_args()
+    return parser.parse_args(args)
+
+
+def _chdir():
+    # smoke test should not be performed on the root directory of checked out source code.
+    import os
+    from pathlib import Path
+
+    os.chdir(Path(__file__).parent)
+    assert "torchaudio" not in os.listdir(os.getcwd())
 
 
 if __name__ == "__main__":

test/smoke_test/smoke_test_no_ffmpeg.py

@@ -0,0 +1,3 @@
+from smoke_test import main
+
+main(["--no-ffmpeg"])