Add test for graph init