+++ /dev/null
-package org.apache.lucene.search.spans;
-
-/**
- * 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.queryParser.QueryParser;
-import org.apache.lucene.search.CheckHits;
-import org.apache.lucene.search.Explanation;
-import org.apache.lucene.search.IndexSearcher;
-import org.apache.lucene.search.Weight;
-import org.apache.lucene.search.Scorer;
-import org.apache.lucene.store.Directory;
-import org.apache.lucene.util.LuceneTestCase;
-
-public class TestNearSpansOrdered extends LuceneTestCase {
- protected IndexSearcher searcher;
- protected Directory directory;
- protected IndexReader reader;
-
- public static final String FIELD = "field";
- public static final QueryParser qp =
- new QueryParser(TEST_VERSION_CURRENT, FIELD, new MockAnalyzer(random));
-
- @Override
- public void tearDown() throws Exception {
- searcher.close();
- reader.close();
- directory.close();
- super.tearDown();
- }
-
- @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()));
- for (int i = 0; i < docFields.length; i++) {
- Document doc = new Document();
- doc.add(newField(FIELD, docFields[i], Field.Store.NO, Field.Index.ANALYZED));
- writer.addDocument(doc);
- }
- reader = writer.getReader();
- writer.close();
- searcher = newSearcher(reader);
- }
-
- protected String[] docFields = {
- "w1 w2 w3 w4 w5",
- "w1 w3 w2 w3 zz",
- "w1 xx w2 yy w3",
- "w1 w3 xx w2 yy w3 zz"
- };
-
- protected SpanNearQuery makeQuery(String s1, String s2, String s3,
- int slop, boolean inOrder) {
- return new SpanNearQuery
- (new SpanQuery[] {
- new SpanTermQuery(new Term(FIELD, s1)),
- new SpanTermQuery(new Term(FIELD, s2)),
- new SpanTermQuery(new Term(FIELD, s3)) },
- slop,
- inOrder);
- }
- protected SpanNearQuery makeQuery() {
- return makeQuery("w1","w2","w3",1,true);
- }
-
- public void testSpanNearQuery() throws Exception {
- SpanNearQuery q = makeQuery();
- CheckHits.checkHits(random, q, FIELD, searcher, new int[] {0,1});
- }
-
- public String s(Spans span) {
- return s(span.doc(), span.start(), span.end());
- }
- public String s(int doc, int start, int end) {
- return "s(" + doc + "," + start + "," + end +")";
- }
-
- public void testNearSpansNext() throws Exception {
- SpanNearQuery q = makeQuery();
- Spans span = q.getSpans(searcher.getIndexReader());
- assertEquals(true, span.next());
- assertEquals(s(0,0,3), s(span));
- assertEquals(true, span.next());
- assertEquals(s(1,0,4), s(span));
- assertEquals(false, span.next());
- }
-
- /**
- * test does not imply that skipTo(doc+1) should work exactly the
- * same as next -- it's only applicable in this case since we know doc
- * does not contain more than one span
- */
- public void testNearSpansSkipToLikeNext() throws Exception {
- SpanNearQuery q = makeQuery();
- Spans span = q.getSpans(searcher.getIndexReader());
- assertEquals(true, span.skipTo(0));
- assertEquals(s(0,0,3), s(span));
- assertEquals(true, span.skipTo(1));
- assertEquals(s(1,0,4), s(span));
- assertEquals(false, span.skipTo(2));
- }
-
- public void testNearSpansNextThenSkipTo() throws Exception {
- SpanNearQuery q = makeQuery();
- Spans span = q.getSpans(searcher.getIndexReader());
- assertEquals(true, span.next());
- assertEquals(s(0,0,3), s(span));
- assertEquals(true, span.skipTo(1));
- assertEquals(s(1,0,4), s(span));
- assertEquals(false, span.next());
- }
-
- public void testNearSpansNextThenSkipPast() throws Exception {
- SpanNearQuery q = makeQuery();
- Spans span = q.getSpans(searcher.getIndexReader());
- assertEquals(true, span.next());
- assertEquals(s(0,0,3), s(span));
- assertEquals(false, span.skipTo(2));
- }
-
- public void testNearSpansSkipPast() throws Exception {
- SpanNearQuery q = makeQuery();
- Spans span = q.getSpans(searcher.getIndexReader());
- assertEquals(false, span.skipTo(2));
- }
-
- public void testNearSpansSkipTo0() throws Exception {
- SpanNearQuery q = makeQuery();
- Spans span = q.getSpans(searcher.getIndexReader());
- assertEquals(true, span.skipTo(0));
- assertEquals(s(0,0,3), s(span));
- }
-
- public void testNearSpansSkipTo1() throws Exception {
- SpanNearQuery q = makeQuery();
- Spans span = q.getSpans(searcher.getIndexReader());
- assertEquals(true, span.skipTo(1));
- assertEquals(s(1,0,4), s(span));
- }
-
- /**
- * not a direct test of NearSpans, but a demonstration of how/when
- * this causes problems
- */
- public void testSpanNearScorerSkipTo1() throws Exception {
- SpanNearQuery q = makeQuery();
- Weight w = searcher.createNormalizedWeight(q);
- Scorer s = w.scorer(searcher.getIndexReader(), true, false);
- assertEquals(1, s.advance(1));
- }
-
- /**
- * not a direct test of NearSpans, but a demonstration of how/when
- * this causes problems
- */
- public void testSpanNearScorerExplain() throws Exception {
- SpanNearQuery q = makeQuery();
- Explanation e = searcher.explain(q, 1);
- assertTrue("Scorer explanation value for doc#1 isn't positive: "
- + e.toString(),
- 0.0f < e.getValue());
- }
-
-}
fnp / pylucene.git / blobdiff