"llama/llama.cpp/examples/llava/clip.cpp" did not exist on "60f75560a2a950e14aabe88c0a7b1223f31277ad"
test_datasets.py 1.4 KB
Newer Older
1
2
3
4
import unittest

from torchaudio.datasets.commonvoice import COMMONVOICE
from torchaudio.datasets.librispeech import LIBRISPEECH
5
from torchaudio.datasets.speechcommands import SPEECHCOMMANDS
Vincent QB's avatar
Vincent QB committed
6
from torchaudio.datasets.utils import diskcache_iterator, bg_iterator
7
8
from torchaudio.datasets.vctk import VCTK
from torchaudio.datasets.yesno import YESNO
Taras Sereda's avatar
Taras Sereda committed
9
from torchaudio.datasets.ljspeech import LJSPEECH
10

Vincent QB's avatar
Vincent QB committed
11
12
import common_utils

13
14

class TestDatasets(unittest.TestCase):
15
    path = common_utils.get_asset_path()
16
17

    def test_yesno(self):
Vincent QB's avatar
Vincent QB committed
18
        data = YESNO(self.path)
19
20
21
        data[0]

    def test_vctk(self):
Vincent QB's avatar
Vincent QB committed
22
        data = VCTK(self.path)
23
24
25
26
27
28
29
        data[0]

    def test_librispeech(self):
        data = LIBRISPEECH(self.path, "dev-clean")
        data[0]

    def test_commonvoice(self):
30
        data = COMMONVOICE(self.path, url="tatar")
31
32
33
        data[0]

    def test_commonvoice_diskcache(self):
34
        data = COMMONVOICE(self.path, url="tatar")
Vincent QB's avatar
Vincent QB committed
35
        data = diskcache_iterator(data)
36
37
38
39
40
        # Save
        data[0]
        # Load
        data[0]

Vincent QB's avatar
Vincent QB committed
41
    def test_commonvoice_bg(self):
42
        data = COMMONVOICE(self.path, url="tatar")
Vincent QB's avatar
Vincent QB committed
43
        data = bg_iterator(data, 5)
moto's avatar
moto committed
44
        for _ in data:
Vincent QB's avatar
Vincent QB committed
45
46
            pass

Taras Sereda's avatar
Taras Sereda committed
47
48
49
50
    def test_ljspeech(self):
        data = LJSPEECH(self.path)
        data[0]

51
52
53
54
    def test_speechcommands(self):
        data = SPEECHCOMMANDS(self.path)
        data[0]

55
56
57

if __name__ == "__main__":
    unittest.main()