Improve testing for the gracedb module