Minor tests changes from comments
Add test for single-class case to PRF
Fix pep8