Add graph unit tests