Add tests for staff grading refactor