Skip to content

Commit 3136b86

Browse files
ahornaceVladimir Kotal
authored andcommitted
Fix failing tests after change to BitSet
1 parent 14f507e commit 3136b86

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

suggester/src/test/java/org/opengrok/suggest/query/customized/CustomSloppyPhraseScorerTest.java

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,9 +38,11 @@
3838
import org.apache.lucene.store.RAMDirectory;
3939
import org.junit.Test;
4040
import org.opengrok.suggest.query.PhraseScorer;
41-
import org.opengrok.suggest.query.data.HashIntsHolder;
41+
import org.opengrok.suggest.query.data.BitIntsHolder;
4242

4343
import java.io.IOException;
44+
import java.util.HashSet;
45+
import java.util.Set;
4446

4547
import static org.hamcrest.collection.IsIterableContainingInOrder.contains;
4648
import static org.junit.Assert.assertThat;
@@ -85,8 +87,20 @@ public static void test(
8587
}
8688
}
8789

88-
assertThat((HashIntsHolder) ((PhraseScorer) scorer).getPositions(correctDoc), contains(expectedPositions));
90+
BitIntsHolder bs = (BitIntsHolder) ((PhraseScorer) scorer).getPositions(correctDoc);
91+
92+
assertThat(toSet(bs), contains(expectedPositions));
93+
}
94+
}
95+
96+
private static Set<Integer> toSet(final BitIntsHolder bs) {
97+
Set<Integer> intSet = new HashSet<>();
98+
for (int i = 0; i < bs.length(); i++) {
99+
if (bs.has(i)) {
100+
intSet.add(i);
101+
}
89102
}
103+
return intSet;
90104
}
91105

92106
@Test

0 commit comments

Comments
 (0)