def test_waves2dataset_defaults():
audiolist = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
train, val, test = ([5, 4, 9, 2, 3, 10, 1, 6], [8], [7])
value_list1, value_list2, value_list3 = featorg.waves2dataset(audiolist)
assert train == value_list1
assert val == value_list2
assert test == value_list3
def test_waves2dataset_seed_0():
audiolist = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
train, val, test = ([7, 1, 2, 5, 6, 9, 10, 8], [4], [3])
value_list1, value_list2, value_list3 = featorg.waves2dataset(audiolist,
seed=0)
assert train == value_list1
assert val == value_list2
assert test == value_list3
def test_waves2dataset_train_perc_50():
audiolist = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
train, val, test = ([5, 4, 9, 2, 3, 10], [1, 6], [8, 7])
value_list1, value_list2, value_list3 = featorg.waves2dataset(
audiolist, train_perc=50)
assert train == value_list1
assert val == value_list2
assert test == value_list3
def test_waves2dataset_seed_none():
audiolist = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
train, val, test = ([7, 1, 2, 5, 6, 9, 10, 8], [4], [3])
value_list1, value_list2, value_list3 = featorg.waves2dataset(audiolist,
seed=None)
assert train != value_list1
assert val != value_list2
assert test != value_list3
assert len(train) == len(value_list1)
assert len(val) == len(value_list2)
assert len(test) == len(value_list3)
def test_waves2dataset_indexerror_str_as_input():
audiolist = 'can this string be separated into datasets?'
with pytest.raises(IndexError):
featorg.waves2dataset(audiolist)
def test_waves2dataset_indexerror_emptylist_as_input():
audiolist = []
with pytest.raises(IndexError):
featorg.waves2dataset(audiolist)