Added tests for graph data preparation