def test_graph_smoke_test():
filename = join(get_test_data_path(), 'image', 'obama.jpg')
stim = ImageStim(filename)
nodes = [(BrightnessExtractor(), 'brightness')]
graph = Graph(nodes)
result = graph.extract([stim])
brightness = result[('BrightnessExtractor', 'brightness')].values[0]
assert_almost_equal(brightness, 0.556134, 5)
def test_small_pipeline():
pytest.importorskip('pytesseract')
filename = join(get_test_data_path(), 'image', 'button.jpg')
stim = ImageStim(filename)
nodes = [(TesseractConverter(), 'tesseract',
[(LengthExtractor(), 'length')])]
graph = Graph(nodes)
result = graph.extract([stim])
assert (0, 'button.jpg_Exit') in result.index
assert ('LengthExtractor', 'text_length') in result.columns
assert result[('LengthExtractor', 'text_length')].values[0] == 4
def test_big_pipeline():
filename = join(get_test_data_path(), 'video', 'obama_speech.mp4')
video = VideoStim(filename)
visual_nodes = [(FrameSamplingConverter(every=15), 'framesampling',
[(TesseractConverter(), 'visual_text',
[(LengthExtractor(), 'text_length')]),
(VibranceExtractor(), 'visual_vibrance')])]
audio_nodes = [(VideoToAudioConverter(), 'audio',
[(WitTranscriptionConverter(), 'audio_text',
[(LengthExtractor(), 'text_length')])])]
graph = Graph()
graph.add_children(visual_nodes)
graph.add_children(audio_nodes)
result = graph.extract(video)
assert ('LengthExtractor', 'text_length') in result.columns
assert ('VibranceExtractor', 'vibrance') in result.columns
assert not result[('onset', '')].isnull().any()
assert 'obama_speech.mp4_obama_speech.wav_today' in result.index.get_level_values(1)
assert 'obama_speech.mp4_90' in result.index.get_level_values(1)
