def test_indico_api_text_extractor():
ext = IndicoAPITextExtractor(api_key=os.environ['INDICO_APP_KEY'],
models=['emotion', 'personality'])
# With ComplexTextStim input
srtfile = join(get_test_data_path(), 'text', 'wonderful.srt')
srt_stim = ComplexTextStim(srtfile, onset=4.2)
result = merge_results(ext.transform(srt_stim), extractor_names=False)
outdfKeysCheck = {
'onset', 'duration', 'order', 'object_id', 'emotion_anger',
'emotion_fear', 'emotion_joy', 'emotion_sadness', 'emotion_surprise',
'personality_openness', 'personality_extraversion',
'personality_agreeableness', 'personality_conscientiousness'
}
meta_columns = {'source_file', 'history', 'class', 'filename'}
assert set(result.columns) - set(['stim_name'
]) == outdfKeysCheck | meta_columns
assert result['onset'][1] == 92.622
# With TextStim input
ts = TextStim(text="It's a wonderful life.")
result = ext.transform(ts).to_df(object_id=True)
assert set(result.columns) == outdfKeysCheck
assert len(result) == 1
def test_indico_api_text_extractor():
ext = IndicoAPITextExtractor(api_key=os.environ['INDICO_APP_KEY'],
models=['emotion', 'personality'])
# With ComplexTextStim input
srtfile = join(get_test_data_path(), 'text', 'wonderful.srt')
srt_stim = ComplexTextStim(srtfile, onset=4.2)
result = ext.transform(srt_stim).to_df()
outdfKeysCheck = set([
'onset',
'duration',
'emotion_anger',
'emotion_fear',
'emotion_joy',
'emotion_sadness',
'emotion_surprise',
'personality_openness',
'personality_extraversion',
'personality_agreeableness',
'personality_conscientiousness'])
assert set(result.columns) == outdfKeysCheck
assert result['onset'][1] == 92.622
# With TextStim input
ts = TextStim(text="It's a wonderful life.")
result = ext.transform(ts).to_df()
assert set(result.columns) == outdfKeysCheck
assert len(result) == 1
def test_indico_api_text_extractor():
ext = IndicoAPITextExtractor(api_key=os.environ['INDICO_APP_KEY'],
models=['emotion', 'personality'])
assert ext.validate_keys()
# With ComplexTextStim input
srtfile = join(get_test_data_path(), 'text', 'wonderful.srt')
srt_stim = ComplexTextStim(srtfile, onset=4.2)
result = merge_results(ext.transform(srt_stim), extractor_names=False)
outdfKeysCheck = {
'onset',
'duration',
'order',
'object_id',
'emotion_anger',
'emotion_fear',
'emotion_joy',
'emotion_sadness',
'emotion_surprise',
'personality_openness',
'personality_extraversion',
'personality_agreeableness',
'personality_conscientiousness'}
meta_columns = {'source_file',
'history',
'class',
'filename'}
assert set(result.columns) - set(['stim_name']) == outdfKeysCheck | meta_columns
assert result['onset'][1] == 92.622
# With TextStim input
ts = TextStim(text="It's a wonderful life.")
result = ext.transform(ts).to_df(object_id=True)
assert set(result.columns) == outdfKeysCheck
assert len(result) == 1
ext = IndicoAPITextExtractor(api_key='nogood', models=['language'])
assert not ext.validate_keys()