Spaces:
Running
Running
File size: 1,445 Bytes
9d61c9b |
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 |
from dataclasses import asdict
from pathlib import Path
import unittest
from training.datasets import HifiGanDataset
class TestHifiGanDataset(unittest.TestCase):
def setUp(self):
self.cache_dir = "datasets_cache"
self.dataset = HifiGanDataset(cache_dir=self.cache_dir, cache=True)
def test_len(self):
# Test that the length of the dataset is correct
self.assertIsInstance(len(self.dataset), int)
def test_get_cache_subdir_path(self):
idx = 1234
expected_path = Path(self.cache_dir) / "cache-hifigan-dataset" / "2000"
self.assertEqual(self.dataset.get_cache_subdir_path(idx), expected_path)
def test_get_cache_file_path(self):
idx = 1234
expected_path = (
Path(self.cache_dir) / "cache-hifigan-dataset" / "2000" / f"{idx}.pt"
)
self.assertEqual(self.dataset.get_cache_file_path(idx), expected_path)
def test_getitem(self):
# Test that getting an item from the dataset returns a HifiGANItem
item = self.dataset[0]
self.assertIsInstance(item, tuple)
def test_iter(self):
# Test that the dataset is iterable
for item in self.dataset:
self.assertIsInstance(item, tuple)
break
def test_cache(self):
cache_file = self.dataset.get_cache_file_path(0)
self.assertTrue(cache_file.exists())
if __name__ == "__main__":
unittest.main()
|