blob: 1db71a47b89554be7629fd66e5fe333f4e41f3f1 (
plain)
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
|
# Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
# SPDX-License-Identifier: MIT
import os
import numpy as np
from context import common_utils
from context import audio_utils
def test_labels(test_data_folder):
labels_file = os.path.join(test_data_folder, "wav2letter_labels.txt")
labels = common_utils.dict_labels(labels_file)
assert len(labels) == 29
assert labels[26] == "\'"
assert labels[27] == r" "
assert labels[28] == "$"
def test_decoder(test_data_folder):
labels_file = os.path.join(test_data_folder, "wav2letter_labels.txt")
labels = common_utils.dict_labels(labels_file)
output_tensor = os.path.join(test_data_folder, "inf_out.npy")
encoded = np.load(output_tensor)
decoded_text = audio_utils.decode(encoded, labels)
assert decoded_text == "and they walkd immediately out of the apartiment by anothe"
|
