package lemmaextractor;

import lemmaextractor.results.Lemma;
import lemmaextractor.results.Result;

import org.fest.assertions.Assertions;
import org.testng.annotations.Test;

public class BasicLemmaExtractorTest {

	private static final String INI_FILE = "/home//mkomar/dane/downloads/LemmaExtractor-pack/LemmaExtractor-pack/test_cfg.ini";
	
	
	@Test(groups="REQUIRE_WCRFT_NATIVE")
	public void shouldLemmatizeText() throws Exception{
		Extractor extractor = new Extractor(INI_FILE);
		Result result = extractor.processFull("Miała kota na dachu");
		Assertions.assertThat(result).isNotNull();
		Assertions.assertThat(result.getLemmas()).hasSize(4);
		Lemma lemma = result.getLemmas().get(0);
		Assertions.assertThat(lemma).isNotNull();
		Assertions.assertThat(lemma.getText().toLowerCase()).isEqualTo("mieć");
		Assertions.assertThat(lemma.getStart()).isEqualTo(0);
		Assertions.assertThat(lemma.getEnd()).isEqualTo(5);
		
		lemma = result.getLemmas().get(1);
		Assertions.assertThat(lemma).isNotNull();
		Assertions.assertThat(lemma.getText().toLowerCase()).isEqualTo("kot");
		Assertions.assertThat(lemma.getStart()).isEqualTo(6);
		Assertions.assertThat(lemma.getEnd()).isEqualTo(10);
		
		lemma = result.getLemmas().get(2);
		Assertions.assertThat(lemma).isNotNull();
		Assertions.assertThat(lemma.getText().toLowerCase()).isEqualTo("na");
		Assertions.assertThat(lemma.getStart()).isEqualTo(11);
		Assertions.assertThat(lemma.getEnd()).isEqualTo(13);
		
		lemma = result.getLemmas().get(3);
		Assertions.assertThat(lemma).isNotNull();
		Assertions.assertThat(lemma.getText().toLowerCase()).isEqualTo("dach");
		Assertions.assertThat(lemma.getStart()).isEqualTo(14);
		Assertions.assertThat(lemma.getEnd()).isEqualTo(19);
		
	}
	
	
}