← Back to branch summary

~slub.team/goobi-indexserver/3.x

« back to all changes in this revision

Viewing changes to lucene/backwards/src/test/org/apache/lucene/index/TestPositionBasedTermVectorMapper.java

  • Committer: Sebastian Meyer
  • Date: 2012-08-03 09:12:40 UTC
  • Revision ID: sebastian.meyer@slub-dresden.de-20120803091240-x6861b0vabq1xror
https://launchpad.net/bugs/985487
Remove Lucene and Solr source code and add patches instead
Fix Bug #985487: Auto-suggestion for the search interface

Show diffs side-by-side

added added

removed removed

Lines of Context:
lucene/backwards/src/test/org/apache/lucene/index/TestPositionBasedTermVectorMapper.java
1
 
package org.apache.lucene.index;
2
 
/**
3
 
 * Copyright 2005 The Apache Software Foundation
4
 
 *
5
 
 * Licensed under the Apache License, Version 2.0 (the "License");
6
 
 * you may not use this file except in compliance with the License.
7
 
 * You may obtain a copy of the License at
8
 
 *
9
 
 *     http://www.apache.org/licenses/LICENSE-2.0
10
 
 *
11
 
 * Unless required by applicable law or agreed to in writing, software
12
 
 * distributed under the License is distributed on an "AS IS" BASIS,
13
 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
 
 * See the License for the specific language governing permissions and
15
 
 * limitations under the License.
16
 
 */
17
 
 
18
 
import org.apache.lucene.util.LuceneTestCase;
19
 
 
20
 
import java.io.IOException;
21
 
import java.util.BitSet;
22
 
import java.util.Map;
23
 
 
24
 
public class TestPositionBasedTermVectorMapper extends LuceneTestCase {
25
 
  protected String[] tokens;
26
 
  protected int[][] thePositions;
27
 
  protected TermVectorOffsetInfo[][] offsets;
28
 
  protected int numPositions;
29
 
 
30
 
  @Override
31
 
  public void setUp() throws Exception {
32
 
    super.setUp();
33
 
    tokens = new String[]{"here", "is", "some", "text", "to", "test", "extra"};
34
 
    thePositions = new int[tokens.length][];
35
 
    offsets = new TermVectorOffsetInfo[tokens.length][];
36
 
    numPositions = 0;
37
 
    //save off the last one so we can add it with the same positions as some of the others, but in a predictable way
38
 
    for (int i = 0; i < tokens.length - 1; i++)
39
 
    {
40
 
      thePositions[i] = new int[2 * i + 1];//give 'em all some positions
41
 
      for (int j = 0; j < thePositions[i].length; j++)
42
 
      {
43
 
        thePositions[i][j] = numPositions++;
44
 
      }
45
 
      offsets[i] = new TermVectorOffsetInfo[thePositions[i].length];
46
 
      for (int j = 0; j < offsets[i].length; j++) {
47
 
        offsets[i][j] = new TermVectorOffsetInfo(j, j + 1);//the actual value here doesn't much matter
48
 
      }
49
 
    }
50
 
    thePositions[tokens.length - 1] = new int[1];
51
 
    thePositions[tokens.length - 1][0] = 0;//put this at the same position as "here"
52
 
    offsets[tokens.length - 1] = new TermVectorOffsetInfo[1];
53
 
    offsets[tokens.length - 1][0] = new TermVectorOffsetInfo(0, 1);
54
 
  }
55
 
 
56
 
  public void test() throws IOException {
57
 
    PositionBasedTermVectorMapper mapper = new PositionBasedTermVectorMapper();
58
 
    
59
 
    mapper.setExpectations("test", tokens.length, true, true);
60
 
    //Test single position
61
 
    for (int i = 0; i < tokens.length; i++) {
62
 
      String token = tokens[i];
63
 
      mapper.map(token, 1, null, thePositions[i]);
64
 
 
65
 
    }
66
 
    Map<String,Map<Integer,PositionBasedTermVectorMapper.TVPositionInfo>> map = mapper.getFieldToTerms();
67
 
    assertTrue("map is null and it shouldn't be", map != null);
68
 
    assertTrue("map Size: " + map.size() + " is not: " + 1, map.size() == 1);
69
 
    Map<Integer,PositionBasedTermVectorMapper.TVPositionInfo> positions = map.get("test");
70
 
    assertTrue("thePositions is null and it shouldn't be", positions != null);
71
 
    
72
 
    assertTrue("thePositions Size: " + positions.size() + " is not: " + numPositions, positions.size() == numPositions);
73
 
    BitSet bits = new BitSet(numPositions);
74
 
    for (Map.Entry<Integer,PositionBasedTermVectorMapper.TVPositionInfo> entry : positions.entrySet()) {
75
 
    
76
 
      PositionBasedTermVectorMapper.TVPositionInfo info = entry.getValue();
77
 
      assertTrue("info is null and it shouldn't be", info != null);
78
 
      int pos = entry.getKey().intValue();
79
 
      bits.set(pos);
80
 
      assertTrue(info.getPosition() + " does not equal: " + pos, info.getPosition() == pos);
81
 
      assertTrue("info.getOffsets() is null and it shouldn't be", info.getOffsets() != null);
82
 
      if (pos == 0)
83
 
      {
84
 
        assertTrue("info.getTerms() Size: " + info.getTerms().size() + " is not: " + 2, info.getTerms().size() == 2);//need a test for multiple terms at one pos
85
 
        assertTrue("info.getOffsets() Size: " + info.getOffsets().size() + " is not: " + 2, info.getOffsets().size() == 2);
86
 
      }
87
 
      else
88
 
      {
89
 
        assertTrue("info.getTerms() Size: " + info.getTerms().size() + " is not: " + 1, info.getTerms().size() == 1);//need a test for multiple terms at one pos
90
 
        assertTrue("info.getOffsets() Size: " + info.getOffsets().size() + " is not: " + 1, info.getOffsets().size() == 1);
91
 
      }
92
 
    }
93
 
    assertTrue("Bits are not all on", bits.cardinality() == numPositions);
94
 
  }
95
 
 
96
 
 
97
 
 
98
 
  
99
 
}