Update grapher mock and tests