The unit test should be isolated. The essential test for this pull request should be the ROC AUC computation. It is hard to it without code refactoring because you put the rocAuc method nested in another method. It would be great if you have time to do the refactoring.