@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); }