Integration test for Classification