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)