def test_bruteforce_test_mat(spect_mat_annot_yarden_durs_labels,
labelset_yarden):
labelset_yarden = set(labelset_yarden)
durs, labels = spect_mat_annot_yarden_durs_labels
train_dur = -1
val_dur = None
test_dur = 200
for _ in range(NUM_SAMPLES):
train_inds, val_inds, test_inds = brute_force(durs, labels,
labelset_yarden,
train_dur, val_dur,
test_dur)
assert is_expected_output(
train_dur,
val_dur,
test_dur,
labelset_yarden,
durs,
labels,
train_inds,
val_inds,
test_inds,
)
def test_bruteforce_test_cbin(audio_cbin_annot_notmat_durs_labels,
labelset_notmat):
labelset_notmat = set(labelset_notmat)
durs, labels = audio_cbin_annot_notmat_durs_labels
train_dur = -1
val_dur = None
test_dur = 25
for _ in range(NUM_SAMPLES):
train_inds, val_inds, test_inds = brute_force(durs, labels,
labelset_notmat,
train_dur, val_dur,
test_dur)
assert is_expected_output(
train_dur,
val_dur,
test_dur,
labelset_notmat,
durs,
labels,
train_inds,
val_inds,
test_inds,
)
def test_bruteforce_test_mat(self):
train_dur = -1
val_dur = None
test_dur = 200
for _ in range(NUM_SAMPLES):
train_inds, val_inds, test_inds = brute_force(
self.durs_mat, self.labels_mat, self.labelset_mat, train_dur,
val_dur, test_dur)
self.assertTrue(
self._check_output(train_dur, val_dur, test_dur,
self.labelset_mat, self.durs_mat,
self.labels_mat, train_inds, val_inds,
test_inds))
def test_bruteforce_train_cbin(self):
train_dur = 35
val_dur = None
test_dur = -1
for _ in range(NUM_SAMPLES):
train_inds, val_inds, test_inds = brute_force(
self.durs_cbin, self.labels_cbin, self.labelset_cbin,
train_dur, val_dur, test_dur)
self.assertTrue(
self._check_output(train_dur, val_dur, test_dur,
self.labelset_cbin, self.durs_cbin,
self.labels_cbin, train_inds, val_inds,
test_inds))
def test_bruteforce_train_test_val_mock(self):
train_dur = 2
test_dur = 2
val_dur = 1
durs = (1, 1, 1, 1, 1)
labelset = set(list('abcde'))
labels = [list('abcde') for _ in range(5)]
for _ in range(NUM_SAMPLES):
train_inds, val_inds, test_inds = brute_force(
durs, labels, labelset, train_dur, val_dur, test_dur)
self.assertTrue(
self._check_output(train_dur, val_dur, test_dur, labelset,
durs, labels, train_inds, val_inds,
test_inds))
def test_bruteforce_train_test_val_mock():
train_dur = 2
test_dur = 2
val_dur = 1
durs = (1, 1, 1, 1, 1)
labelset = set(list("abcde"))
labels = [list("abcde") for _ in range(5)]
for _ in range(NUM_SAMPLES):
train_inds, val_inds, test_inds = brute_force(durs, labels, labelset,
train_dur, val_dur,
test_dur)
assert is_expected_output(
train_dur,
val_dur,
test_dur,
labelset,
durs,
labels,
train_inds,
val_inds,
test_inds,
)
