Skip to content

Commit 9d1cefd

Browse files
JRMeyerloganhart02
authored andcommitted
first pass read flac from file
1 parent bb75afb commit 9d1cefd

File tree

6 files changed

+198
-100
lines changed

6 files changed

+198
-100
lines changed

.github/workflows/lint.yml

+2
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,9 @@ jobs:
1818
python-version: ${{ matrix.pyver }}
1919
- name: Run training unittests
2020
run: |
21+
sudo apt-get install -y --no-install-recommends libopusfile0 libopus-dev libopusfile-dev
2122
./ci_scripts/train-unittests.sh
23+
2224
pre-commit-checks:
2325
name: "Lin|Pre-commit checks"
2426
runs-on: ubuntu-20.04

Dockerfile.build

+1
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
1919
libbz2-dev \
2020
libboost-all-dev \
2121
libgsm1-dev \
22+
libopusfile0 \
2223
libltdl-dev \
2324
liblzma-dev \
2425
libmagic-dev \

Dockerfile.train

+1
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ RUN apt-get update && \
3535
git \
3636
wget \
3737
libopus0 \
38+
libvorbisfile3 \
3839
libopusfile0 \
3940
libsndfile1 \
4041
sox \

tests/Example.ogg

103 KB
Binary file not shown.

tests/test_audio_utils.py

+54
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
1+
import unittest
2+
from argparse import Namespace
3+
from pathlib import Path
4+
5+
from coqui_stt_training.util import audio
6+
7+
8+
def from_here(path):
9+
here = Path(__file__)
10+
return here.parent / path
11+
12+
13+
class TestValidateReadAudio(unittest.TestCase):
14+
def test_flac(self):
15+
audio_path = from_here("../data/smoke_test/LDC93S1.flac")
16+
audio_type = audio.get_loadable_audio_type_from_extension(audio_path.suffix)
17+
result = audio.read_audio(audio_type, str(audio_path))
18+
self.assertIsInstance(result[0], audio.AudioFormat)
19+
20+
def test_wav(self):
21+
audio_path = from_here("../data/smoke_test/LDC93S1.wav")
22+
audio_type = audio.get_loadable_audio_type_from_extension(audio_path.suffix)
23+
result = audio.read_audio(audio_type, str(audio_path))
24+
self.assertIsInstance(result[0], audio.AudioFormat)
25+
26+
def test_ogg_opus(self):
27+
audio_path = from_here("../data/smoke_test/LDC93S1.opus")
28+
audio_type = audio.get_loadable_audio_type_from_extension(audio_path.suffix)
29+
result = audio.read_audio(audio_type, str(audio_path))
30+
self.assertIsInstance(result[0], audio.AudioFormat)
31+
32+
33+
class TestValidateReadDuration(unittest.TestCase):
34+
def test_flac(self):
35+
audio_path = from_here("../data/smoke_test/LDC93S1.flac")
36+
audio_type = audio.get_loadable_audio_type_from_extension(audio_path.suffix)
37+
result = audio.read_duration(audio_type, str(audio_path))
38+
self.assertIsInstance(result, float)
39+
40+
def test_wav(self):
41+
audio_path = from_here("../data/smoke_test/LDC93S1.wav")
42+
audio_type = audio.get_loadable_audio_type_from_extension(audio_path.suffix)
43+
result = audio.read_duration(audio_type, str(audio_path))
44+
self.assertIsInstance(result, float)
45+
46+
def test_ogg_opus(self):
47+
audio_path = from_here("../data/smoke_test/LDC93S1.opus")
48+
audio_type = audio.get_loadable_audio_type_from_extension(audio_path.suffix)
49+
result = audio.read_duration(audio_type, str(audio_path))
50+
self.assertIsInstance(result, float)
51+
52+
53+
if __name__ == "__main__":
54+
unittest.main()

0 commit comments

Comments
 (0)