Add tests for BreadthFirstModel helper methods