Move unit test helper from main graph