from src.graph_builder import GraphBuilder

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