/** * Initialize the {@link NodeDisjunctionScorerQueue}. */ private NodeDisjunctionScorerQueue initNodeScorerQueue() throws IOException { final NodeDisjunctionScorerQueue nodeQueue = new NodeDisjunctionScorerQueue(nrScorers); for (final NodeScorer s : scorers) { nodeQueue.put(s); } return nodeQueue; }
@Test public void testNextNodeHeapTraversal() throws IOException { this.addDocument("\"term1 term3\" \"term5 term2\" . \"term1 term3\" . \"term5 term4 term3\" . "); final NodeDisjunctionScorerQueue q = new NodeDisjunctionScorerQueue(4); q.put(this.getScorer(ntq("term2"))); q.put(this.getScorer(ntq("term3"))); q.put(this.getScorer(ntq("term4"))); q.put(this.getScorer(ntq("term5"))); // test if the heap traversal is done properly. for (int i = 0; i < 4; i++) { q.nextNodeAndAdjust(); } assertFalse(q.nextNodeAndAdjust()); }
@Test public void testScoreSum() throws IOException { this.addDocument("\"term1 term2 term3\" . \"term4\" . "); final NodeScorer s1 = this.getScorer(ntq("term1")); final NodeScorer s2 = this.getScorer(ntq("term2")); final NodeScorer s3 = this.getScorer(ntq("term3")); final NodeDisjunctionScorerQueue q = new NodeDisjunctionScorerQueue(3); q.put(s1); q.put(s2); q.put(s3); assertEquals(0, q.doc()); assertTrue(q.nextNodeAndAdjust()); q.countAndSumMatchers(); // there must be 3 matchers assertEquals(3, q.nrMatchersInNode()); final float scoreInNode = q.scoreInNode(); assertEquals(s1.scoreInNode() + s2.scoreInNode() + s3.scoreInNode(), scoreInNode, 0f); assertTrue(scoreInNode != 0); }
/** * Test method for {@link org.sindice.siren.search.node.NodeDisjunctionScorerQueue#top()}. * @throws IOException * @throws CorruptIndexException */ @Test public void testTop() throws IOException { this.addDocument("\"term1\" . "); this.addDocument("\"term2\" . "); this.addDocument("\"term3\" . \"term4\" . "); final NodeDisjunctionScorerQueue q = new NodeDisjunctionScorerQueue(5); final NodeScorer s1 = this.getScorer(ntq("term1")); final NodeScorer s2 = this.getScorer(ntq("term2")); final NodeScorer s3 = this.getScorer(ntq("term3")); final NodeScorer s4 = this.getScorer(ntq("term4")); q.put(s3); assertSame(s3, q.top()); q.put(s2); assertSame(s2, q.top()); // s2 should be the least scorer q.put(s4); assertSame(s2, q.top()); q.put(s1); assertSame(s1, q.top()); // s1 should be the least scorer }
@Test public void testNextCandidateDocumentAndAdjustElsePop() throws IOException { this.addDocument("\"term1\" \"term2\" . \"term3\" . \"term4\" . "); this.addDocument("\"term5\" \"term2\" . \"term3\" . "); final NodeDisjunctionScorerQueue q = new NodeDisjunctionScorerQueue(4); q.put(this.getScorer(ntq("term2"))); q.put(this.getScorer(ntq("term3"))); q.put(this.getScorer(ntq("term4"))); q.put(this.getScorer(ntq("term5"))); assertEquals(0, q.doc()); assertEquals(4, q.size()); assertTrue(q.nextCandidateDocumentAndAdjustElsePop()); assertEquals(1, q.doc()); assertEquals(3, q.size()); // term4 scorer should have been removed assertFalse(q.nextCandidateDocumentAndAdjustElsePop()); assertEquals(DocsAndNodesIterator.NO_MORE_DOC, q.doc()); assertEquals(DocsAndNodesIterator.NO_MORE_NOD, q.node()); }
@Test public void testNrMatches() throws IOException { this.addDocument("\"term1\" \"term2\" . \"term3\" . \"term4\" . "); this.addDocument("\"term2\" \"term3\" . \"term5\" . "); this.addDocument("\"term2\" \"term1 term5\" . "); final NodeDisjunctionScorerQueue q = new NodeDisjunctionScorerQueue(2); q.put(this.getScorer(ntq("term1"))); q.put(this.getScorer(ntq("term5"))); assertEquals(0, q.doc()); assertTrue(q.nextNodeAndAdjust()); q.countAndSumMatchers(); // there must be 1 matchers for node {0,0} assertEquals(1, q.nrMatchersInNode()); assertTrue(q.nextCandidateDocumentAndAdjustElsePop()); assertEquals(1, q.doc()); assertTrue(q.nextNodeAndAdjust()); q.countAndSumMatchers(); // there must be 1 matchers for node {1,0} assertEquals(1, q.nrMatchersInNode()); assertTrue(q.nextCandidateDocumentAndAdjustElsePop()); assertEquals(2, q.doc()); assertTrue(q.nextNodeAndAdjust()); q.countAndSumMatchers(); // there must be 2 matchers for node {0,1} assertEquals(2, q.nrMatchersInNode()); }
this.addDocument("\"term2\" \"term1\" . \"term5\" . "); final NodeDisjunctionScorerQueue q = new NodeDisjunctionScorerQueue(2); q.put(this.getScorer(ntq("term1"))); q.put(this.getScorer(ntq("term5")));
this.addDocument("\"term3\" . \"term3\" . "); final NodeDisjunctionScorerQueue q = new NodeDisjunctionScorerQueue(3); q.put(this.getScorer(ntq("term1"))); q.put(this.getScorer(ntq("term2")));