Apache
/
lucene
mirror of https://github.com/apache/lucene.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

LUCENE-5493: simplify this test 

git-svn-id: https://svn.apache.org/repos/asf/lucene/dev/branches/lucene5493@1574945 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Robert Muir 2014-03-06 16:22:12 +00:00
parent f39f4f66c1
commit 44e2e3155f
1 changed files with 11 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
lucene/misc/src/test/org/apache/lucene/index/sorter/SortingAtomicReaderTest.java
View File

@ -17,49 +17,37 @@ package org.apache.lucene.index.sorter;
* limitations under the License. * limitations under the License.
*/ */
import java.io.IOException;
import java.util.Arrays; import java.util.Arrays;
import org.apache.lucene.index.AtomicReader; import org.apache.lucene.index.NumericDocValues;
import org.apache.lucene.search.Sort;
import org.apache.lucene.search.SortField; import org.apache.lucene.search.SortField;
import org.apache.lucene.util.Bits; import org.apache.lucene.util.Bits;
import org.apache.lucene.util.TestUtil; import org.apache.lucene.util.TestUtil;
import org.apache.lucene.util.TestUtil;
import org.junit.BeforeClass; import org.junit.BeforeClass;
public class SortingAtomicReaderTest extends SorterTestBase { public class SortingAtomicReaderTest extends SorterTestBase {
@BeforeClass @BeforeClass
public static void beforeClassSortingAtomicReaderTest() throws Exception { public static void beforeClassSortingAtomicReaderTest() throws Exception {
// build the mapping from the reader, since we deleted documents, some of
// them might have disappeared from the index (e.g. if an entire segment is
// dropped b/c all its docs are deleted)
final int[] values = new int[reader.maxDoc()];
for (int i = 0; i < reader.maxDoc(); i++) {
values[i] = Integer.valueOf(reader.document(i).get(ID_FIELD));
}
final Sorter.DocComparator comparator = new Sorter.DocComparator() {
@Override
public int compare(int docID1, int docID2) {
final int v1 = values[docID1];
final int v2 = values[docID2];
return v1 < v2 ? -1 : v1 == v2 ? 0 : 1;
}
};
final Sorter.DocMap docMap = Sorter.sort(reader.maxDoc(), comparator); // sort the index by id (as integer, in NUMERIC_DV_FIELD)
Sort sort = new Sort(new SortField(NUMERIC_DV_FIELD, SortField.Type.INT));
final Sorter.DocMap docMap = new SortSorter(sort).sort(reader);
// Sorter.compute also sorts the values // Sorter.compute also sorts the values
NumericDocValues dv = reader.getNumericDocValues(NUMERIC_DV_FIELD);
sortedValues = new Integer[reader.maxDoc()]; sortedValues = new Integer[reader.maxDoc()];
for (int i = 0; i < reader.maxDoc(); ++i) { for (int i = 0; i < reader.maxDoc(); ++i) {
sortedValues[docMap.oldToNew(i)] = values[i]; sortedValues[docMap.oldToNew(i)] = (int)dv.get(i);
} }
if (VERBOSE) { if (VERBOSE) {
System.out.println("docMap: " + docMap); System.out.println("docMap: " + docMap);
System.out.println("sortedValues: " + Arrays.toString(sortedValues)); System.out.println("sortedValues: " + Arrays.toString(sortedValues));
} }
// TODO: what is this doing? like a no-op sort? // sort the index by id (as integer, in NUMERIC_DV_FIELD)
reader = SortingAtomicReader.wrap(reader, docMap); reader = SortingAtomicReader.wrap(reader, sort);
if (VERBOSE) { if (VERBOSE) {
System.out.print("mapped-deleted-docs: "); System.out.print("mapped-deleted-docs: ");