D-Net

package eu.dnetlib.pace.clustering;

import java.util.Map;

import org.junit.Before;

import org.junit.Test;

import com.google.common.collect.Lists;

import com.google.common.collect.Maps;

import eu.dnetlib.pace.AbstractPaceTest;

/**

 * The Class ClusteringFunctionTest.

 */

public class ClusteringFunctionTest extends AbstractPaceTest {

	/** The params. */

	private Map<String, Integer> params;

	/**

	 * Sets the up.

	 * 

	 * @throws Exception

	 *             the exception

	 */

	@Before

	public void setUp() throws Exception {

		params = Maps.newHashMap();

	}

	/**

	 * Test ngram.

	 */

	@Test

	public void testNgram() {

		params.put("ngramLen", 2);

		params.put("max", 8);

		params.put("maxPerToken", 1);

		params.put("minNgramLen", 3);

		ClusteringFunction ngram = new Ngrams(params);

		String s = "Search for the Standard Model Higgs Boson";

		System.out.println(s);

		System.out.println(ngram.apply(Lists.newArrayList(title(s))));

	}

	/**

	 * Test ngram pairs.

	 */

	@Test

	public void testNgramPairs() {

		params.put("ngramLen", 2);

		params.put("max", 4);

		ClusteringFunction np = new NgramPairs(params);

		String s = "Search for the Standard Model Higgs Boson";

		System.out.println(s);

		System.out.println(np.apply(Lists.newArrayList(title(s))));

	}

	/**

	 * Test acronym.

	 */

	@Test

	public void testAcronym() {

		params.put("max", 4);

		params.put("minLen", 1);

		params.put("maxLen", 3);

		ClusteringFunction acro = new Acronyms(params);

		String s = "Search for the Standard Model Higgs Boson";

		System.out.println(s);

		System.out.println(acro.apply(Lists.newArrayList(title(s))));

	}

	/**

	 * Test suffix prefix.

	 */

	@Test

	public void testSuffixPrefix() {

		params.put("len", 2);

		params.put("max", 3);

		ClusteringFunction sp = new SuffixPrefix(params);

		String s = "Search for the Standard Model Higgs Boson";

		System.out.println(s);

		System.out.println(sp.apply(Lists.newArrayList(title(s))));

	}

}