AGDISTISTest.java

import static org.junit.Assert.assertTrue;

import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Properties;

import org.aksw.agdistis.algorithm.CandidateUtil;
import org.aksw.agdistis.algorithm.NEDAlgo_HITS;
import org.aksw.agdistis.datatypes.Document;
import org.aksw.agdistis.datatypes.NamedEntitiesInText;
import org.aksw.agdistis.datatypes.NamedEntityInText;
import org.aksw.agdistis.util.TripleIndexContext;
import org.aksw.agdistis.webapp.GetDisambiguation;
import org.junit.Test;

public class AGDISTISTest {

	@Test
	public void testUmlaute() throws InterruptedException, IOException {
		String taisho = "Emperor Taishō";
		String taishoURL = "http://dbpedia.org/resource/Emperor_Taishō";
		String japan = "Japan";
		String japanURL = "http://dbpedia.org/resource/Japan";

		HashMap<String, String> correct = new HashMap<String, String>();
		correct.put(taisho, taishoURL);
		correct.put(japan, japanURL);
		String preAnnotatedText = "<entity>" + taisho + "</entity> was the 123rd Emperor of <entity>" + japan
				+ "</entity>.";

		NEDAlgo_HITS agdistis = new NEDAlgo_HITS();
		Document d = GetDisambiguation.textToDocument(preAnnotatedText);
		agdistis.run(d, null);

		NamedEntitiesInText namedEntities = d.getNamedEntitiesInText();
		HashMap<NamedEntityInText, String> results = new HashMap<NamedEntityInText, String>();
		for (NamedEntityInText namedEntity : namedEntities) {
			String disambiguatedURL = namedEntity.getNamedEntityUri();
			results.put(namedEntity, disambiguatedURL);
		}
		for (NamedEntityInText namedEntity : namedEntities) {
			String disambiguatedURL = namedEntity.getNamedEntityUri();
			System.out.println(namedEntity.getLabel() + " -> " + disambiguatedURL);
			assertTrue(correct.get(namedEntity.getLabel()).equals(disambiguatedURL));
		}

	}

	@Test
	public void testMinimalExample() throws InterruptedException, IOException {
		String obama = "Barack Obama";
		String obamaURL = "http://dbpedia.org/resource/Barack_Obama";
		String merkel = "Angela Merkel";
		String merkelURL = "http://dbpedia.org/resource/Angela_Merkel";
		String city = "Berlin";
		String cityURL = "http://dbpedia.org/resource/Berlin";

		HashMap<String, String> correct = new HashMap<String, String>();
		correct.put(obama, obamaURL);
		correct.put(merkel, merkelURL);
		correct.put(city, cityURL);

		String preAnnotatedText = "<entity>" + obama + "</entity> visits <entity>" + merkel + "</entity> in <entity>"
				+ city + "</entity>.";

		NEDAlgo_HITS agdistis = new NEDAlgo_HITS();
		Document d = GetDisambiguation.textToDocument(preAnnotatedText);
		agdistis.run(d, null);

		NamedEntitiesInText namedEntities = d.getNamedEntitiesInText();
		HashMap<NamedEntityInText, String> results = new HashMap<NamedEntityInText, String>();
		for (NamedEntityInText namedEntity : namedEntities) {
			String disambiguatedURL = namedEntity.getNamedEntityUri();
			results.put(namedEntity, disambiguatedURL);
		}
		for (NamedEntityInText namedEntity : namedEntities) {
			String disambiguatedURL = namedEntity.getNamedEntityUri();
			System.out.println(namedEntity.getLabel() + " -> " + disambiguatedURL);
			assertTrue(correct.get(namedEntity.getLabel()).equals(disambiguatedURL));
		}

	}

	@Test
	/**
	 * This test ensures that entities consiting only of punctations, which are
	 * later reduced to a string of length 0 by
	 * https://github.com/dice-group/AGDISTIS/blob/master/src/main/java/org/aksw/agdistis/util/Stemming.java#L91
	 * are not causing exceptions
	 * 
	 * @throws InterruptedException
	 * @throws IOException
	 */
	public void testinterpunctation() throws InterruptedException, IOException {
		String question = "???";
		String questionURL = "???";

		HashMap<String, String> correct = new HashMap<String, String>();
		correct.put(question, questionURL);

		String preAnnotatedText = "<entity>???</entity>.";

		NEDAlgo_HITS agdistis = new NEDAlgo_HITS();
		Document d = GetDisambiguation.textToDocument(preAnnotatedText);
		agdistis.run(d, null);

		NamedEntitiesInText namedEntities = d.getNamedEntitiesInText();
		HashMap<NamedEntityInText, String> results = new HashMap<NamedEntityInText, String>();
		for (NamedEntityInText namedEntity : namedEntities) {
			String disambiguatedURL = namedEntity.getNamedEntityUri();
			results.put(namedEntity, disambiguatedURL);
		}
		for (NamedEntityInText namedEntity : namedEntities) {
			String disambiguatedURL = namedEntity.getNamedEntityUri();
			System.out.println(namedEntity.getLabel() + " -> " + disambiguatedURL);
			assertTrue(correct.get(namedEntity.getLabel()).equals(disambiguatedURL));
		}

	}

	@Test
	public void testContext() throws InterruptedException, IOException {
		String angelina = "Angelina";
		String angelinaURL = "http://dbpedia.org/resource/Angelina_Jolie";
		String brad = "Brad";
		String bradURL = "http://dbpedia.org/resource/Brad_Pitt";
		String jon = "Jon";
		String jonURL = "http://dbpedia.org/resource/Jon_Voight";

		// load properties to see if the context index exists
		Properties prop = new Properties();
		InputStream input = CandidateUtil.class.getResourceAsStream("/config/agdistis.properties");
		prop.load(input);
		String envContext = System.getenv("AGDISTIS_CONTEXT");
		Boolean context = Boolean.valueOf(envContext != null ? envContext : prop.getProperty("context"));
		if (context == true) { // in case the index by context exist

			HashMap<String, String> correct = new HashMap<String, String>();
			correct.put(angelina, angelinaURL);
			correct.put(jon, jonURL);
			correct.put(brad, bradURL);

			String preAnnotatedText = "<entity>" + angelina + "</entity>, her father <entity>" + jon
					+ "</entity>, and her partner <entity>" + brad
					+ "</entity> never played together in the same movie.";

			NEDAlgo_HITS agdistis = new NEDAlgo_HITS();
			Document d = GetDisambiguation.textToDocument(preAnnotatedText);
			agdistis.run(d, null);

			NamedEntitiesInText namedEntities = d.getNamedEntitiesInText();
			HashMap<NamedEntityInText, String> results = new HashMap<NamedEntityInText, String>();
			for (NamedEntityInText namedEntity : namedEntities) {
				String disambiguatedURL = namedEntity.getNamedEntityUri();
				results.put(namedEntity, disambiguatedURL);
			}
			for (NamedEntityInText namedEntity : namedEntities) {
				String disambiguatedURL = namedEntity.getNamedEntityUri();
				System.out.println(namedEntity.getLabel() + " -> " + disambiguatedURL);
				assertTrue(correct.get(namedEntity.getLabel()).equals(disambiguatedURL));
			}
		}
	}

}