Coverage for graphqler / fuzzer / utils.py: 0%

1import networkx 

2 

3from graphqler.graph import Node 

4 

5 

6def get_node(graph: networkx.DiGraph, name: str) -> Node: 

7 """Gets a node from the graph with the same name 

8 

9 Args: 

10 graph (networkx.DiGraph): The graph to look in 

11 name (str): The name of the node 

12 

13 Returns: 

14 Node: A node matching the name. Raises exception if node isn't found 

15 """ 

16 for node in graph.nodes: 

17 if node.name == name: 

18 return node 

19 raise Exception(f"Node with name {name} not found in graph")