Add test for graph models