def test_has_arrowhead_with_marked_arrowhead(): graph = MarkedPatternGraph(nodes=['a', 'b']) graph.add_undirected_edge(('a', 'b')) assert graph.has_arrowhead(('a', 'b')) == False graph.add_marked_arrowhead(('a', 'b')) assert graph.has_arrowhead(('a', 'b')) == True assert graph.has_marked_arrowhead(('a', 'b')) == True
def test_has_marked_path(): graph = MarkedPatternGraph(nodes=['a', 'b']) graph.add_undirected_edge(('a', 'b')) graph.add_marked_arrowhead(('a', 'b')) assert graph.has_marked_arrowhead(('a', 'b')) == True graph.add_undirected_edge(('b', 'c')) assert graph.has_marked_path(('a', 'c')) == False graph.add_marked_arrowhead(('b', 'c')) assert graph.has_marked_path(('a', 'c')) == True assert graph.get_nodes_of_edges() == set({'a', 'b', 'c'})