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, )