articleclassifiertest.java

来自「一个简单的kNN算法实现」· Java 代码 · 共 39 行

package article;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.junit.Test;

import article.entity.Article;
import article.entity.Category;
import article.service.impl.ArticleClassifierImpl;

/**
 * @author ahuaxuan(aaron zhang)
 * @since 2008-2-18
 * @version $Id$
 */
public class ArticleClassifierTest {

	@Test
	public void testClassifier() {
		ArticleClassifierImpl classifier = new ArticleClassifierImpl();
		Map<String, List<String>> map = classifier.matchArticle(prepareCategoryList(), prepareArticleList());
		System.out.println(map.size());
	}
	
	public static List<Category> prepareCategoryList() {
		List<Category> categoryList = new ArrayList<Category>();
		List<Article> articleList = new ArrayList<Article>();
		
		Category c = new Category();
		c.setArticleList(articleList);
		c.setId("1");
		categoryList.add(c);
		
		Article a1 = new Article();
		a1.setId("1");
		a1.setTitle("Hibernate鍒濇帰");
		a1.setContent("寮