Adds tests for ControlGraph

Issue: #32