File size: 1,620 Bytes
2cba4ca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import sys
import unittest
from test_transcribe import *
import test_transcribe
if __name__ == '__main__':
# Handle several ways of generating expected outputs
if "--long" in sys.argv:
test_transcribe.SKIP_LONG_TEST_IF_CPU = False
sys.argv.remove("--long")
if "--generate" in sys.argv:
test_transcribe.FAIL_IF_REFERENCE_NOT_FOUND = False
sys.argv.remove("--generate")
if "--generate_device" in sys.argv:
test_transcribe.GENERATE_DEVICE_DEPENDENT = True
test_transcribe.FAIL_IF_REFERENCE_NOT_FOUND = False
sys.argv.remove("--generate_device")
if "--generate_new" in sys.argv:
test_transcribe.GENERATE_NEW_ONLY = True
test_transcribe.FAIL_IF_REFERENCE_NOT_FOUND = False
sys.argv.remove("--generate_new")
if "--generate_all" in sys.argv:
test_transcribe.GENERATE_ALL = True
test_transcribe.FAIL_IF_REFERENCE_NOT_FOUND = False
sys.argv.remove("--generate_all")
# Pass options to whisper_timestamped CLI
args = sys.argv[1:]
for i, arg in enumerate(args):
if arg not in [
"-h", "--help",
"-v", "--verbose",
"--locals",
"-q", "--quiet",
"-f", "--failfast",
"-c", "--catch",
"-b", "--buffer",
"-k",
] \
and not arg.startswith("Test") \
and (i==0 or args[i-1] not in ["-k"]) \
and (arg.startswith("-") or (i>0 and args[i-1].startswith("-"))):
test_transcribe.CMD_OPTIONS.append(arg)
sys.argv.remove(arg)
unittest.main()
|