fix test and use model to easily test the graph contents