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