File size: 701 Bytes
0b32ad6 |
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 |
import pytest
from dotenv import dotenv_values
from s3prl.dataio.corpus.snips import SNIPS
@pytest.mark.corpus
def test_snips():
config = dotenv_values()
dataset_root = config["SNIPS"]
dataset = SNIPS(
dataset_root,
[
"Ivy",
"Joanna",
"Joey",
"Justin",
"Kendra",
"Kimberly",
"Matthew",
"Salli",
],
["Aditi", "Amy", "Geraint", "Nicole"],
["Brian", "Emma", "Raveena", "Russell"],
)
train_data, valid_data, test_data = dataset.data_split
assert len(train_data) == 104672
assert len(valid_data) == 2800
assert len(test_data) == 2800
|