+++ /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.analysis.MockAnalyzer;
-import org.apache.lucene.document.Document;
-import org.apache.lucene.document.Field;
-import org.apache.lucene.index.IndexReader;
-import org.apache.lucene.index.RandomIndexWriter;
-import org.apache.lucene.index.Term;
-import org.apache.lucene.search.BooleanClause.Occur;
-import org.apache.lucene.store.Directory;
-import org.apache.lucene.util.DocIdBitSet;
-import org.apache.lucene.util.LuceneTestCase;
-
-import java.util.BitSet;
-
-/**
- * FilteredQuery JUnit tests.
- *
- * <p>Created: Apr 21, 2004 1:21:46 PM
- *
- *
- * @since 1.4
- */
-public class TestFilteredQuery extends LuceneTestCase {
-
- private IndexSearcher searcher;
- private IndexReader reader;
- private Directory directory;
- private Query query;
- private Filter filter;
-
- @Override
- public void setUp() throws Exception {
- super.setUp();
- directory = newDirectory();
- RandomIndexWriter writer = new RandomIndexWriter (random, directory, newIndexWriterConfig(TEST_VERSION_CURRENT, new MockAnalyzer(random)).setMergePolicy(newLogMergePolicy()));
-
- Document doc = new Document();
- doc.add (newField("field", "one two three four five", Field.Store.YES, Field.Index.ANALYZED));
- doc.add (newField("sorter", "b", Field.Store.YES, Field.Index.ANALYZED));
- writer.addDocument (doc);
-
- doc = new Document();
- doc.add (newField("field", "one two three four", Field.Store.YES, Field.Index.ANALYZED));
- doc.add (newField("sorter", "d", Field.Store.YES, Field.Index.ANALYZED));
- writer.addDocument (doc);
-
- doc = new Document();
- doc.add (newField("field", "one two three y", Field.Store.YES, Field.Index.ANALYZED));
- doc.add (newField("sorter", "a", Field.Store.YES, Field.Index.ANALYZED));
- writer.addDocument (doc);
-
- doc = new Document();
- doc.add (newField("field", "one two x", Field.Store.YES, Field.Index.ANALYZED));
- doc.add (newField("sorter", "c", Field.Store.YES, Field.Index.ANALYZED));
- writer.addDocument (doc);
-
- // tests here require single segment (eg try seed
- // 8239472272678419952L), because SingleDocTestFilter(x)
- // blindly accepts that docID in any sub-segment
- writer.optimize();
-
- reader = writer.getReader();
- writer.close ();
-
- searcher = newSearcher(reader);
- query = new TermQuery (new Term ("field", "three"));
- filter = newStaticFilterB();
- }
-
- // must be static for serialization tests
- private static Filter newStaticFilterB() {
- return new Filter() {
- @Override
- public DocIdSet getDocIdSet (IndexReader reader) {
- BitSet bitset = new BitSet(5);
- bitset.set (1);
- bitset.set (3);
- return new DocIdBitSet(bitset);
- }
- };
- }
-
- @Override
- public void tearDown() throws Exception {
- searcher.close();
- reader.close();
- directory.close();
- super.tearDown();
- }
-
- public void testFilteredQuery()
- throws Exception {
- Query filteredquery = new FilteredQuery (query, filter);
- ScoreDoc[] hits = searcher.search (filteredquery, null, 1000).scoreDocs;
- assertEquals (1, hits.length);
- assertEquals (1, hits[0].doc);
- QueryUtils.check(random, filteredquery,searcher);
-
- hits = searcher.search (filteredquery, null, 1000, new Sort(new SortField("sorter", SortField.STRING))).scoreDocs;
- assertEquals (1, hits.length);
- assertEquals (1, hits[0].doc);
-
- filteredquery = new FilteredQuery (new TermQuery (new Term ("field", "one")), filter);
- hits = searcher.search (filteredquery, null, 1000).scoreDocs;
- assertEquals (2, hits.length);
- QueryUtils.check(random, filteredquery,searcher);
-
- filteredquery = new FilteredQuery (new TermQuery (new Term ("field", "x")), filter);
- hits = searcher.search (filteredquery, null, 1000).scoreDocs;
- assertEquals (1, hits.length);
- assertEquals (3, hits[0].doc);
- QueryUtils.check(random, filteredquery,searcher);
-
- filteredquery = new FilteredQuery (new TermQuery (new Term ("field", "y")), filter);
- hits = searcher.search (filteredquery, null, 1000).scoreDocs;
- assertEquals (0, hits.length);
- QueryUtils.check(random, filteredquery,searcher);
-
- // test boost
- Filter f = newStaticFilterA();
-
- float boost = 2.5f;
- BooleanQuery bq1 = new BooleanQuery();
- TermQuery tq = new TermQuery (new Term ("field", "one"));
- tq.setBoost(boost);
- bq1.add(tq, Occur.MUST);
- bq1.add(new TermQuery (new Term ("field", "five")), Occur.MUST);
-
- BooleanQuery bq2 = new BooleanQuery();
- tq = new TermQuery (new Term ("field", "one"));
- filteredquery = new FilteredQuery(tq, f);
- filteredquery.setBoost(boost);
- bq2.add(filteredquery, Occur.MUST);
- bq2.add(new TermQuery (new Term ("field", "five")), Occur.MUST);
- assertScoreEquals(bq1, bq2);
-
- assertEquals(boost, filteredquery.getBoost(), 0);
- assertEquals(1.0f, tq.getBoost(), 0); // the boost value of the underlying query shouldn't have changed
- }
-
- // must be static for serialization tests
- private static Filter newStaticFilterA() {
- return new Filter() {
- @Override
- public DocIdSet getDocIdSet (IndexReader reader) {
- BitSet bitset = new BitSet(5);
- bitset.set(0, 5);
- return new DocIdBitSet(bitset);
- }
- };
- }
-
- /**
- * Tests whether the scores of the two queries are the same.
- */
- public void assertScoreEquals(Query q1, Query q2) throws Exception {
- ScoreDoc[] hits1 = searcher.search (q1, null, 1000).scoreDocs;
- ScoreDoc[] hits2 = searcher.search (q2, null, 1000).scoreDocs;
-
- assertEquals(hits1.length, hits2.length);
-
- for (int i = 0; i < hits1.length; i++) {
- assertEquals(hits1[i].score, hits2[i].score, 0.0000001f);
- }
- }
-
- /**
- * This tests FilteredQuery's rewrite correctness
- */
- public void testRangeQuery() throws Exception {
- TermRangeQuery rq = new TermRangeQuery(
- "sorter", "b", "d", true, true);
-
- Query filteredquery = new FilteredQuery(rq, filter);
- ScoreDoc[] hits = searcher.search(filteredquery, null, 1000).scoreDocs;
- assertEquals(2, hits.length);
- QueryUtils.check(random, filteredquery,searcher);
- }
-
- public void testBoolean() throws Exception {
- BooleanQuery bq = new BooleanQuery();
- Query query = new FilteredQuery(new MatchAllDocsQuery(),
- new SingleDocTestFilter(0));
- bq.add(query, BooleanClause.Occur.MUST);
- query = new FilteredQuery(new MatchAllDocsQuery(),
- new SingleDocTestFilter(1));
- bq.add(query, BooleanClause.Occur.MUST);
- ScoreDoc[] hits = searcher.search(bq, null, 1000).scoreDocs;
- assertEquals(0, hits.length);
- QueryUtils.check(random, query,searcher);
- }
-
- // Make sure BooleanQuery, which does out-of-order
- // scoring, inside FilteredQuery, works
- public void testBoolean2() throws Exception {
- BooleanQuery bq = new BooleanQuery();
- Query query = new FilteredQuery(bq,
- new SingleDocTestFilter(0));
- bq.add(new TermQuery(new Term("field", "one")), BooleanClause.Occur.SHOULD);
- bq.add(new TermQuery(new Term("field", "two")), BooleanClause.Occur.SHOULD);
- ScoreDoc[] hits = searcher.search(query, 1000).scoreDocs;
- assertEquals(1, hits.length);
- QueryUtils.check(random, query,searcher);
- }
-}
-
-
-
-
fnp / pylucene.git / blobdiff