tests/test_graph_building.py

from src.graph_builder import GraphBuilder

def test_graph():
    builder = GraphBuilder()
    graph = builder.build(['a','b'])
    assert len(graph.nodes) == 2