def test_add_nested_children(): graph = Graph() de1, de2, de3 = DummyExtractor(), DummyExtractor(), DummyExtractor() graph.add_children([de1, (de2, 'child', [(de3, "child's child")])]) assert len(graph.children) == 2 assert isinstance(graph.children[1].children[0], Node) assert graph.children[1].children[0].name == "child's child"
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)
def test_add_children(): graph = Graph() de1, de2, de3 = DummyExtractor(), DummyExtractor(), DummyExtractor() graph.add_children([de1, de2, de3]) assert len(graph.children) == 3 assert all([isinstance(c, Node) for c in graph.children])