def test_renderer(self): filename = os.path.dirname(__file__) + '/sample_files/rend-file.xml' comparation_filename = os.path.dirname(__file__) + \ '/sample_files/expected-rend-file.xml' from_node = Node('node_zero') node = Node('node_one') anchors = ('1', '2') region = Region('region_one', *anchors) node.add_region(region) features = {'name': 'feature_name', 'value': 'feature_value'} annotation = Annotation('Annotation_label', features, 'annotation-1') node.annotations.add(annotation) edge = Edge('edge1', from_node, node) self.graph.edges.add(edge) self.graph.nodes.add(node) self.graph.regions.add(region) graf_render = GrafRenderer(filename) graf_render.render(self.graph) expected_tree = ElementTree.parse(comparation_filename) result_tree = ElementTree.parse(filename) expected_result = [ElementTree.tostring(i) for i in expected_tree.getroot()] result = [ElementTree.tostring(i) for i in result_tree.getroot()] assert (result[0] == expected_result[0]) assert (result[1] == expected_result[1]) assert (result[2] == expected_result[2]) assert (result[3] == expected_result[3])
def test_renderer(self): filename = os.path.dirname(__file__) + '/sample_files/rend-file.xml' comparation_filename = os.path.dirname(__file__) + \ '/sample_files/expected-rend-file.xml' from_node = Node('node_zero') node = Node('node_one') anchors = ('1', '2') region = Region('region_one', *anchors) node.add_region(region) features = {'name': 'feature_name', 'value': 'feature_value'} annotation = Annotation('Annotation_label', features, 'annotation-1') node.annotations.add(annotation) edge = Edge('edge1', from_node, node) self.graph.edges.add(edge) self.graph.nodes.add(node) self.graph.regions.add(region) graf_render = GrafRenderer(filename) graf_render.render(self.graph) expected_tree = ElementTree.parse(comparation_filename) result_tree = ElementTree.parse(filename) expected_result = [ ElementTree.tostring(i) for i in expected_tree.getroot() ] result = [ElementTree.tostring(i) for i in result_tree.getroot()] assert (result[0] == expected_result[0]) assert (result[1] == expected_result[1]) assert (result[2] == expected_result[2]) assert (result[3] == expected_result[3])
graph.nodes.add(node_two) graph.nodes.add(node_three) # Create the edge edge = Edge(node_one, node_three) # Add an the edge #graph.add_edge(edge) # Add Features feature_strct = FeatureStructure() # Structure # Adding two features # There are two ways to it. # Use the simple first # Setting up all the features to the annotation feature_strct['feature1'] = 'value_1' feature_strct['feature2'] = 'value_2' # Adding the features to annotations annotation = Annotation('label', feature_strct) annotation.features['another_feature'] = 'value_another' annotation.features['feature3'] = 'value_3' # Adding the annotations to the node node_one.annotations.add(annotation) # Rendering the Graph graf_render = GrafRenderer(sys.stdout) graf_render.render(graph)