add unit test for classifier