test(compute_statistics): convert to python

Author: eginhard

TTS/bin/compute_statistics.py

@@ -5,6 +5,7 @@
 import glob
 import logging
 import os
+from typing import Optional
 
 import numpy as np
 from tqdm import tqdm
@@ -16,10 +17,7 @@
 from TTS.utils.generic_utils import ConsoleFormatter, setup_logger
 
 
-def main():
-    """Run preprocessing process."""
-    setup_logger("TTS", level=logging.INFO, screen=True, formatter=ConsoleFormatter())
-
+def parse_args(arg_list: Optional[list[str]]) -> tuple[argparse.Namespace, list[str]]:
     parser = argparse.ArgumentParser(description="Compute mean and variance of spectrogtram features.")
     parser.add_argument("config_path", type=str, help="TTS config file path to define audio processin parameters.")
     parser.add_argument("out_path", type=str, help="save path (directory and filename).")
@@ -29,7 +27,13 @@ def main():
         required=False,
         help="folder including the target set of wavs overriding dataset config.",
     )
-    args, overrides = parser.parse_known_args()
+    return parser.parse_known_args(arg_list)
+
+
+def main(arg_list: Optional[list[str]] = None):
+    """Run preprocessing process."""
+    setup_logger("TTS", level=logging.INFO, screen=True, formatter=ConsoleFormatter())
+    args, overrides = parse_args(arg_list)
 
     CONFIG = load_config(args.config_path)
     CONFIG.parse_known_args(overrides, relaxed_parser=True)

run_bash_tests.sh

@@ -3,5 +3,4 @@ TF_CPP_MIN_LOG_LEVEL=3
 
 # runtime bash based tests
 # TODO: move these to python
-./tests/bash_tests/test_demo_server.sh && \
-./tests/bash_tests/test_compute_statistics.sh
+./tests/bash_tests/test_demo_server.sh

tests/aux_tests/test_compute_statistics.py

@@ -0,0 +1,12 @@
+from pathlib import Path
+
+from tests import get_tests_input_path
+from TTS.bin.compute_statistics import main
+
+
+def test_compute_statistics(tmp_path):
+    config_path = Path(get_tests_input_path()) / "test_glow_tts_config.json"
+    output_path = tmp_path / "scale_stats.npy"
+
+    args = ["--config_path", str(config_path), "--out_path", str(output_path)]
+    main(args)