def _get_ref_mos(dataset_reader, mos): ref_mos = [] for dis_video in dataset_reader.dataset.dis_videos: # get the dis video's ref video's mos curr_content_id = dis_video['content_id'] ref_indices = indices( zip(dataset_reader.content_id_of_dis_videos, dataset_reader.disvideo_is_refvideo), lambda content_ref_pair: content_ref_pair[ 1] and content_ref_pair[0] == curr_content_id) assert len(ref_indices) == 1, \ 'Should have only and one ref video for a dis video, ' \ 'but got {}'.format(len(ref_indices)) ref_idx = ref_indices[0] ref_mos.append(mos[ref_idx]) return np.array(ref_mos)
def test_disvideo_is_refvideo(self): l = self.dataset_reader.disvideo_is_refvideo self.assertItemsEqual(indices(l, lambda e: e is True), range(7))