Testing with sklean like setup