forked from MontrealCorpusTools/Montreal-Forced-Aligner
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_commandline_validate.py
74 lines (64 loc) · 1.75 KB
/
test_commandline_validate.py
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
import os
import sys
import pytest
from montreal_forced_aligner.command_line.mfa import parser
from montreal_forced_aligner.command_line.validate import run_validate_corpus
def test_validate_corpus(
multilingual_ipa_tg_corpus_dir, english_mfa_acoustic_model, english_us_mfa_dictionary, temp_dir
):
if sys.platform == "win32":
pytest.skip("Transcription testing uses ngram")
command = [
"validate",
multilingual_ipa_tg_corpus_dir,
english_us_mfa_dictionary,
english_mfa_acoustic_model,
"-t",
temp_dir,
"-q",
"--clean",
"--disable_mp",
"--test_transcriptions",
]
args, unknown = parser.parse_known_args(command)
run_validate_corpus(args)
def test_validate_training_corpus(
multilingual_ipa_tg_corpus_dir,
english_dictionary,
temp_dir,
mono_train_config_path,
):
command = [
"validate",
multilingual_ipa_tg_corpus_dir,
english_dictionary,
"-t",
os.path.join(temp_dir, "validation"),
"-q",
"--clean",
"--config_path",
mono_train_config_path,
]
args, unknown = parser.parse_known_args(command)
run_validate_corpus(args)
@pytest.mark.skip("Outdated models")
def test_validate_missing_phones(
multilingual_ipa_tg_corpus_dir,
german_prosodylab_acoustic_model,
german_prosodylab_dictionary,
temp_dir,
):
command = [
"validate",
multilingual_ipa_tg_corpus_dir,
german_prosodylab_dictionary,
german_prosodylab_acoustic_model,
"-t",
temp_dir,
"-q",
"--clean",
"--debug",
"--ignore_acoustics",
]
args, unknown = parser.parse_known_args(command)
run_validate_corpus(args)