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])
