--- /dev/null
+package org.apache.lucene.search;
+
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import org.apache.lucene.util.LuceneTestCase;
+import org.apache.lucene.analysis.standard.StandardAnalyzer;
+import org.apache.lucene.document.Document;
+import org.apache.lucene.document.Field;
+import org.apache.lucene.index.IndexWriter;
+import org.apache.lucene.queryParser.ParseException;
+import org.apache.lucene.queryParser.QueryParser;
+import org.apache.lucene.store.Directory;
+import java.io.IOException;
+
+/**
+ * Tests {@link MultiSearcher} ranking, i.e. makes sure this bug is fixed:
+ * http://issues.apache.org/bugzilla/show_bug.cgi?id=31841
+ *
+ */
+public class TestMultiSearcherRanking extends LuceneTestCase {
+
+ private final String FIELD_NAME = "body";
+ private Searcher multiSearcher;
+ private Searcher singleSearcher;
+
+ public void testOneTermQuery() throws IOException, ParseException {
+ checkQuery("three");
+ }
+
+ public void testTwoTermQuery() throws IOException, ParseException {
+ checkQuery("three foo");
+ }
+
+ public void testPrefixQuery() throws IOException, ParseException {
+ checkQuery("multi*");
+ }
+
+ public void testFuzzyQuery() throws IOException, ParseException {
+ checkQuery("multiThree~");
+ }
+
+ public void testRangeQuery() throws IOException, ParseException {
+ checkQuery("{multiA TO multiP}");
+ }
+
+ public void testMultiPhraseQuery() throws IOException, ParseException {
+ checkQuery("\"blueberry pi*\"");
+ }
+
+ public void testNoMatchQuery() throws IOException, ParseException {
+ checkQuery("+three +nomatch");
+ }
+
+ /*
+ public void testTermRepeatedQuery() throws IOException, ParseException {
+ // TODO: this corner case yields different results.
+ checkQuery("multi* multi* foo");
+ }
+ */
+
+ /**
+ * checks if a query yields the same result when executed on
+ * a single IndexSearcher containing all documents and on a
+ * MultiSearcher aggregating sub-searchers
+ * @param queryStr the query to check.
+ * @throws IOException
+ * @throws ParseException
+ */
+ private void checkQuery(String queryStr) throws IOException, ParseException {
+ // check result hit ranking
+ if(VERBOSE) System.out.println("Query: " + queryStr);
+ QueryParser queryParser = new QueryParser(TEST_VERSION_CURRENT, FIELD_NAME, new StandardAnalyzer(TEST_VERSION_CURRENT));
+ Query query = queryParser.parse(queryStr);
+ ScoreDoc[] multiSearcherHits = multiSearcher.search(query, null, 1000).scoreDocs;
+ ScoreDoc[] singleSearcherHits = singleSearcher.search(query, null, 1000).scoreDocs;
+ assertEquals(multiSearcherHits.length, singleSearcherHits.length);
+ for (int i = 0; i < multiSearcherHits.length; i++) {
+ Document docMulti = multiSearcher.doc(multiSearcherHits[i].doc);
+ Document docSingle = singleSearcher.doc(singleSearcherHits[i].doc);
+ if(VERBOSE) System.out.println("Multi: " + docMulti.get(FIELD_NAME) + " score="
+ + multiSearcherHits[i].score);
+ if(VERBOSE) System.out.println("Single: " + docSingle.get(FIELD_NAME) + " score="
+ + singleSearcherHits[i].score);
+ assertEquals(multiSearcherHits[i].score, singleSearcherHits[i].score,
+ 0.001f);
+ assertEquals(docMulti.get(FIELD_NAME), docSingle.get(FIELD_NAME));
+ }
+ if(VERBOSE) System.out.println();
+ }
+
+ /**
+ * initializes multiSearcher and singleSearcher with the same document set
+ */
+ @Override
+ public void setUp() throws Exception {
+ super.setUp();
+ // create MultiSearcher from two seperate searchers
+ d1 = newDirectory();
+ IndexWriter iw1 = new IndexWriter(d1, newIndexWriterConfig(TEST_VERSION_CURRENT, new StandardAnalyzer(TEST_VERSION_CURRENT)).setMergePolicy(newLogMergePolicy()));
+ addCollection1(iw1);
+ iw1.close();
+ d2 = newDirectory();
+ IndexWriter iw2 = new IndexWriter(d2, newIndexWriterConfig(TEST_VERSION_CURRENT, new StandardAnalyzer(TEST_VERSION_CURRENT)).setMergePolicy(newLogMergePolicy()));
+ addCollection2(iw2);
+ iw2.close();
+
+ Searchable[] s = new Searchable[2];
+ s[0] = new IndexSearcher(d1, true);
+ s[1] = new IndexSearcher(d2, true);
+ multiSearcher = new MultiSearcher(s);
+
+ // create IndexSearcher which contains all documents
+ d = newDirectory();
+ IndexWriter iw = new IndexWriter(d, newIndexWriterConfig(TEST_VERSION_CURRENT, new StandardAnalyzer(TEST_VERSION_CURRENT)).setMergePolicy(newLogMergePolicy()));
+ addCollection1(iw);
+ addCollection2(iw);
+ iw.close();
+ singleSearcher = new IndexSearcher(d, true);
+ }
+
+ Directory d1, d2, d;
+
+ @Override
+ public void tearDown() throws Exception {
+ multiSearcher.close();
+ singleSearcher.close();
+ d1.close();
+ d2.close();
+ d.close();
+ super.tearDown();
+ }
+
+ private void addCollection1(IndexWriter iw) throws IOException {
+ add("one blah three", iw);
+ add("one foo three multiOne", iw);
+ add("one foobar three multiThree", iw);
+ add("blueberry pie", iw);
+ add("blueberry strudel", iw);
+ add("blueberry pizza", iw);
+ }
+
+ private void addCollection2(IndexWriter iw) throws IOException {
+ add("two blah three", iw);
+ add("two foo xxx multiTwo", iw);
+ add("two foobar xxx multiThreee", iw);
+ add("blueberry chewing gum", iw);
+ add("bluebird pizza", iw);
+ add("bluebird foobar pizza", iw);
+ add("piccadilly circus", iw);
+ }
+
+ private void add(String value, IndexWriter iw) throws IOException {
+ Document d = new Document();
+ d.add(newField(FIELD_NAME, value, Field.Store.YES, Field.Index.ANALYZED));
+ iw.addDocument(d);
+ }
+
+}
fnp / pylucene.git / blobdiff