Add knn classifier test API