add test for graph_builder