Add graph tests