@Test
public void testSimilarity() {
LexRankRanker summarizer = new LexRankRanker();
Sentence sentence1 = text.getSentences().get(0);
Sentence sentence2 = text.getSentences().get(1);
Map<String, Double> idf = summarizer.inverseDocumentFrequency(text);
double similarity = summarizer.similarity(sentence1, sentence2, idf);
Assert.assertEquals(0.2340, similarity, 0.0001);
}
@Test
public void testInverseDocumentFrequency() {
LexRankRanker summarizer = new LexRankRanker();
Map<String, Double> wordFrequencies = summarizer.inverseDocumentFrequency(text);
Assert.assertEquals(1.0116, wordFrequencies.get("not"), 0.0001);
Assert.assertEquals(1.7047, wordFrequencies.get("president"), 0.0001);
Assert.assertEquals(1.2992, wordFrequencies.get("refuses"), 0.0001);
Assert.assertEquals(2.3978, wordFrequencies.get("strike"), 0.0001);
}
