17 lines
392 B
Python
17 lines
392 B
Python
|
from plesna.graph import Edge, Graph, Node
|
||
|
|
|
||
|
|
|
||
|
|
def test_init():
|
||
|
|
nodeA = Node(name="A")
|
||
|
|
nodeB = Node(name="B")
|
||
|
|
nodeC = Node(name="C")
|
||
|
|
|
||
|
|
edge1 = Edge(arrow_name="arrow", source=nodeA, target=nodeC)
|
||
|
|
edge2 = Edge(arrow_name="arrow", source=nodeB, target=nodeC)
|
||
|
|
|
||
|
|
graph = Graph()
|
||
|
|
graph.append(edge1)
|
||
|
|
graph.append(edge2)
|
||
|
|
|
||
|
|
assert graph.nodes == {nodeA, nodeB, nodeC}
|