summaryrefslogtreecommitdiffstats
path: root/predicate-search/src/test/java/com/yahoo/search/predicate/index/BoundsPostingListTest.java
diff options
context:
space:
mode:
Diffstat (limited to 'predicate-search/src/test/java/com/yahoo/search/predicate/index/BoundsPostingListTest.java')
-rw-r--r--predicate-search/src/test/java/com/yahoo/search/predicate/index/BoundsPostingListTest.java84
1 files changed, 84 insertions, 0 deletions
diff --git a/predicate-search/src/test/java/com/yahoo/search/predicate/index/BoundsPostingListTest.java b/predicate-search/src/test/java/com/yahoo/search/predicate/index/BoundsPostingListTest.java
new file mode 100644
index 00000000000..06782e3603a
--- /dev/null
+++ b/predicate-search/src/test/java/com/yahoo/search/predicate/index/BoundsPostingListTest.java
@@ -0,0 +1,84 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.search.predicate.index;
+
+import com.google.common.primitives.Ints;
+import org.junit.Test;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import static java.util.stream.Collectors.toList;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * This test lies in com.yahoo.search.predicate to get access to some methods of PredicateIndex.
+ *
+ * @author <a href="mailto:magnarn@yahoo-inc.com">Magnar Nedland</a>
+ */
+public class BoundsPostingListTest {
+
+ @Test
+ public void requireThatPostingListChecksBounds() {
+ PredicateIntervalStore.Builder builder = new PredicateIntervalStore.Builder();
+ List<Integer> docIds = new ArrayList<>();
+ List<Integer> dataRefs = new ArrayList<>();
+ for (int id = 1; id < 100; ++id) {
+ List<IntervalWithBounds> boundsList = new ArrayList<>();
+ for (int i = 0; i <= id; ++i) {
+ int bounds;
+ if (id < 30) {
+ bounds = 0x80000000 | i; // diff >= i
+ } else if (id < 60) {
+ bounds = 0x40000000 | i; // diff < i
+ } else {
+ bounds = (i << 16) | (i + 10); // i < diff < i + 10
+ }
+ boundsList.add(new IntervalWithBounds((i + 1) << 16 | 0xffff, bounds));
+ }
+ docIds.add(id);
+ dataRefs.add(builder.insert(boundsList.stream().flatMap(IntervalWithBounds::stream).collect(toList())));
+ }
+
+ PredicateIntervalStore store = builder.build();
+ BoundsPostingList postingList = new BoundsPostingList(
+ store, Ints.toArray(docIds), Ints.toArray(dataRefs), 0xffffffffffffffffL, 5);
+ assertEquals(-1, postingList.getDocId());
+ assertEquals(0, postingList.getInterval());
+ assertEquals(0xffffffffffffffffL, postingList.getSubquery());
+
+ checkNext(postingList, 0, 1, 2); // [0..] .. [1..]
+ checkNext(postingList, 1, 2, 3); // [0..] .. [2..]
+ checkNext(postingList, 10, 11, 6); // [0..] .. [5..]
+ checkNext(postingList, 20, 21, 6);
+
+ checkNext(postingList, 30, 31, 26); // [..5] .. [..30]
+ checkNext(postingList, 50, 51, 46);
+
+ checkNext(postingList, 60, 61, 6); // [0..10] .. [5..15]
+
+ postingList = new BoundsPostingList(store, Ints.toArray(docIds), Ints.toArray(dataRefs), 0xffffffffffffffffL, 40);
+ checkNext(postingList, 0, 1, 2);
+ checkNext(postingList, 20, 21, 22);
+
+ checkNext(postingList, 30, 31, 0); // skip ahead to match
+ checkNext(postingList, 32, 33, 0); // skip ahead to match
+ checkNext(postingList, 33, 34, 0); // skip ahead to match
+ checkNext(postingList, 40, 41, 1);
+ checkNext(postingList, 50, 51, 11); // [..40] .. [..50]
+
+ checkNext(postingList, 60, 61, 10); // [31..40] .. [40..49]
+ }
+
+ private void checkNext(BoundsPostingList postingList, int movePast, int docId, int intervalCount) {
+ assertTrue("Unable to move past " + movePast, postingList.nextDocument(movePast));
+ assertEquals(intervalCount > 0, postingList.prepareIntervals());
+ assertEquals(docId, postingList.getDocId());
+ for (int i = 0; i < intervalCount - 1; ++i) {
+ assertTrue("Too few intervals, expected " + intervalCount, postingList.nextInterval());
+ }
+ assertFalse("Too many intervals, expected " + intervalCount, postingList.nextInterval());
+ }
+
+}