test code for sklearn