def test_klab_Zhang2018search(): stimulus_set = brainio.get_stimulus_set('klab.Zhang2018.search_obj_array') # There are 300 presentation images in the assembly but 606 in the StimulusSet (explanation from @shashikg follows). # For each of the visual search task out of total 300, you need two images (one - the target image, # second - the search space image) plus there are 6 different mask images to mask objects # present at 6 different locations in a specified search image. # Therefore, a total of 300 * 2 + 6 images are there in the stimulus set. assert len(stimulus_set) == 606 assert len(set(stimulus_set['image_id'])) == 606
def test_feifei_Deng2009(): stimulus_set = brainio.get_stimulus_set('fei-fei.Deng2009') assert len(stimulus_set) == 50_000 assert len(set(stimulus_set['label'])) == 1_000
def test_digital(self): stimulus_set = brainio.get_stimulus_set( 'dietterich.Hendrycks2019.digital') assert len(stimulus_set) == 4 * 5 * 50000 assert len(set(stimulus_set['synset'])) == 1000
def test_noise(self): stimulus_set = brainio.get_stimulus_set( 'dietterich.Hendrycks2019.noise') assert len(stimulus_set) == 3 * 5 * 50000 assert len(set(stimulus_set['synset'])) == 1000
def test_num_stimuli(self, identifier, num_stimuli): stimulus_set = brainio.get_stimulus_set(identifier) assert len(stimulus_set) == num_stimuli