def test_save_graph():
graph = Graph(
spec=join(get_test_data_path(), 'graph', 'simple_graph.json'))
filename = tempfile.mkstemp()[1]
graph.save(filename)
assert os.path.exists(filename)
same_graph = Graph(spec=filename)
os.remove(filename)
assert graph.to_json() == same_graph.to_json()
img = join(get_test_data_path(), 'image', 'button.jpg')
res = same_graph.run(img)
assert res['LengthExtractor#text_length'][0] == 4
def test_to_json():
nodes = {
"roots": [{
"transformer":
"FrameSamplingFilter",
"parameters": {
"every": 15
},
"children": [{
"transformer": "TesseractConverter",
"children": [{
"transformer": "LengthExtractor"
}]
}, {
"transformer": "VibranceExtractor"
}, {
"transformer": "BrightnessExtractor"
}]
}, {
"transformer":
"VideoToAudioConverter",
"children": [{
"transformer": "WitTranscriptionConverter",
"children": [{
"transformer": "LengthExtractor"
}]
}]
}]
}
graph = Graph(nodes)
assert graph.to_json() == nodes
graph = Graph(
spec=join(get_test_data_path(), 'graph', 'simple_graph.json'))
simple_graph = {
"roots": [{
"transformer": "TesseractConverter",
"children": [{
"transformer": "LengthExtractor"
}]
}]
}
assert graph.to_json() == simple_graph
filename = join(get_test_data_path(), 'image', 'button.jpg')
res = graph.run(filename)
assert res['LengthExtractor#text_length'][0] == 4
