Finish output test for stratified cox
Fix input tests