Skip to content

Commit 263d809

Browse files
committed
A few more
1 parent 7125cf7 commit 263d809

File tree

6 files changed

+41
-25
lines changed

6 files changed

+41
-25
lines changed

lucene/core/src/test/org/apache/lucene/search/TestSearcherManager.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
*/
1717
package org.apache.lucene.search;
1818

19+
import static org.hamcrest.Matchers.instanceOf;
20+
1921
import java.io.IOException;
2022
import java.util.ArrayList;
2123
import java.util.List;
@@ -531,9 +533,9 @@ public void testCustomDirectoryReader() throws Exception {
531533
mgr.maybeRefresh();
532534
IndexSearcher s = mgr.acquire();
533535
try {
534-
assertTrue(s.getIndexReader() instanceof MyFilterDirectoryReader);
536+
assertThat(s.getIndexReader(), instanceOf(MyFilterDirectoryReader.class));
535537
for (LeafReaderContext ctx : s.getIndexReader().leaves()) {
536-
assertTrue(ctx.reader() instanceof MyFilterLeafReader);
538+
assertThat(ctx.reader(), instanceOf(MyFilterLeafReader.class));
537539
}
538540
} finally {
539541
mgr.release(s);

lucene/core/src/test/org/apache/lucene/search/TestTermScorer.java

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,12 @@
1616
*/
1717
package org.apache.lucene.search;
1818

19+
import static org.hamcrest.Matchers.equalTo;
20+
import static org.hamcrest.Matchers.greaterThan;
21+
import static org.hamcrest.Matchers.hasSize;
22+
import static org.hamcrest.Matchers.instanceOf;
23+
import static org.hamcrest.Matchers.oneOf;
24+
1925
import java.io.IOException;
2026
import java.util.ArrayList;
2127
import java.util.List;
@@ -86,7 +92,7 @@ public void test() throws IOException {
8692
TermQuery termQuery = new TermQuery(allTerm);
8793

8894
Weight weight = indexSearcher.createWeight(termQuery, ScoreMode.COMPLETE, 1);
89-
assertTrue(indexSearcher.getTopReaderContext() instanceof LeafReaderContext);
95+
assertThat(indexSearcher.getTopReaderContext(), instanceOf(LeafReaderContext.class));
9096
LeafReaderContext context = (LeafReaderContext) indexSearcher.getTopReaderContext();
9197
BulkScorer ts = weight.bulkScorer(context);
9298
// we have 2 documents with the term all in them, one document for all the
@@ -109,9 +115,8 @@ public void collect(int doc) throws IOException {
109115
float score = scorer.score();
110116
doc = doc + base;
111117
docs.add(new TestHit(doc, score));
112-
assertTrue("score " + score + " is not greater than 0", score > 0);
113-
assertTrue(
114-
"Doc: " + doc + " does not equal 0 or doc does not equal 5", doc == 0 || doc == 5);
118+
assertThat(score, greaterThan(0f));
119+
assertThat(doc, oneOf(0, 5));
115120
}
116121

117122
@Override
@@ -127,11 +132,11 @@ public ScoreMode scoreMode() {
127132
null,
128133
0,
129134
DocIdSetIterator.NO_MORE_DOCS);
130-
assertTrue("docs Size: " + docs.size() + " is not: " + 2, docs.size() == 2);
135+
assertThat(docs, hasSize(2));
131136
TestHit doc0 = docs.get(0);
132137
TestHit doc5 = docs.get(1);
133138
// The scores should be the same
134-
assertTrue(doc0.score + " does not equal: " + doc5.score, doc0.score == doc5.score);
139+
assertEquals(doc5.score, doc0.score, 0);
135140
}
136141

137142
public void testNext() throws Exception {
@@ -140,7 +145,7 @@ public void testNext() throws Exception {
140145
TermQuery termQuery = new TermQuery(allTerm);
141146

142147
Weight weight = indexSearcher.createWeight(termQuery, ScoreMode.COMPLETE, 1);
143-
assertTrue(indexSearcher.getTopReaderContext() instanceof LeafReaderContext);
148+
assertThat(indexSearcher.getTopReaderContext(), instanceOf(LeafReaderContext.class));
144149
LeafReaderContext context = (LeafReaderContext) indexSearcher.getTopReaderContext();
145150
Scorer ts = weight.scorer(context);
146151
assertTrue(
@@ -158,12 +163,12 @@ public void testAdvance() throws Exception {
158163
TermQuery termQuery = new TermQuery(allTerm);
159164

160165
Weight weight = indexSearcher.createWeight(termQuery, ScoreMode.COMPLETE, 1);
161-
assertTrue(indexSearcher.getTopReaderContext() instanceof LeafReaderContext);
166+
assertThat(indexSearcher.getTopReaderContext(), instanceOf(LeafReaderContext.class));
162167
LeafReaderContext context = (LeafReaderContext) indexSearcher.getTopReaderContext();
163168
Scorer ts = weight.scorer(context);
164169
assertTrue("Didn't skip", ts.iterator().advance(3) != DocIdSetIterator.NO_MORE_DOCS);
165170
// The next doc should be doc 5
166-
assertTrue("doc should be number 5", ts.docID() == 5);
171+
assertThat(ts.docID(), equalTo(5));
167172
}
168173

169174
private static class TestHit {

lucene/core/src/test/org/apache/lucene/search/TestTopDocsMerge.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
*/
1717
package org.apache.lucene.search;
1818

19+
import static org.hamcrest.Matchers.greaterThan;
20+
1921
import java.io.IOException;
2022
import java.util.ArrayList;
2123
import java.util.Collections;
@@ -124,9 +126,9 @@ public void testPreAssignedShardIndex() {
124126
// that we already set, instead of the position of that TopDocs in the array:
125127
TopDocs merge = TopDocs.merge(from, size, topDocs.toArray(new TopDocs[0]));
126128

127-
assertTrue(merge.scoreDocs.length > 0);
129+
assertThat(merge.scoreDocs.length, greaterThan(0));
128130
for (ScoreDoc scoreDoc : merge.scoreDocs) {
129-
assertTrue(scoreDoc.shardIndex != -1);
131+
assertNotEquals(-1, scoreDoc.shardIndex);
130132
TopDocs shardTopDocs = shardResultMapping.get(scoreDoc.shardIndex);
131133
assertNotNull(shardTopDocs);
132134
boolean found = false;

lucene/core/src/test/org/apache/lucene/search/TestTopFieldCollector.java

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,9 @@
1717
package org.apache.lucene.search;
1818

1919
import static org.apache.lucene.search.SortField.FIELD_SCORE;
20+
import static org.hamcrest.Matchers.greaterThan;
21+
import static org.hamcrest.Matchers.lessThanOrEqualTo;
22+
import static org.hamcrest.Matchers.sameInstance;
2023

2124
import java.io.IOException;
2225
import java.util.Arrays;
@@ -560,7 +563,7 @@ public void testConcurrentMinScore() throws Exception {
560563
new TopFieldCollectorManager(sort, 2, 0);
561564
TopFieldCollector collector = manager.newCollector();
562565
TopFieldCollector collector2 = manager.newCollector();
563-
assertTrue(collector.minScoreAcc == collector2.minScoreAcc);
566+
assertThat(collector2.minScoreAcc, sameInstance(collector.minScoreAcc));
564567
MaxScoreAccumulator minValueChecker = collector.minScoreAcc;
565568
// force the check of the global minimum score on every round
566569
minValueChecker.modInterval = 0;
@@ -691,8 +694,8 @@ public void testRandomMinCompetitiveScore() throws Exception {
691694
TopDocs tdc = doConcurrentSearchWithThreshold(5, 0, query, sort, indexReader);
692695
TopDocs tdc2 = doSearchWithThreshold(5, 0, query, sort, indexReader);
693696

694-
assertTrue(tdc.totalHits.value() > 0);
695-
assertTrue(tdc2.totalHits.value() > 0);
697+
assertThat(tdc.totalHits.value(), greaterThan(0L));
698+
assertThat(tdc2.totalHits.value(), greaterThan(0L));
696699
CheckHits.checkEqual(query, tdc.scoreDocs, tdc2.scoreDocs);
697700
}
698701

@@ -721,7 +724,7 @@ public void testRelationVsTopDocsCount() throws Exception {
721724

722725
collectorManager = new TopFieldCollectorManager(sort, 2, null, 2);
723726
topDocs = searcher.search(new TermQuery(new Term("f", "foo")), collectorManager);
724-
assertTrue(10 >= topDocs.totalHits.value());
727+
assertThat(topDocs.totalHits.value(), lessThanOrEqualTo(10L));
725728
assertEquals(TotalHits.Relation.GREATER_THAN_OR_EQUAL_TO, topDocs.totalHits.relation());
726729

727730
collectorManager = new TopFieldCollectorManager(sort, 10, null, 2);

lucene/core/src/test/org/apache/lucene/search/TestWANDScorer.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,9 @@
1616
*/
1717
package org.apache.lucene.search;
1818

19+
import static org.hamcrest.Matchers.greaterThanOrEqualTo;
20+
import static org.hamcrest.Matchers.lessThan;
21+
1922
import java.io.IOException;
2023
import java.util.Arrays;
2124
import java.util.List;
@@ -63,8 +66,8 @@ public void testScalingFactor() {
6366
private void doTestScalingFactor(float f) {
6467
int scalingFactor = WANDScorer.scalingFactor(f);
6568
float scaled = Math.scalb(f, scalingFactor);
66-
assertTrue("" + scaled, scaled >= 1 << (WANDScorer.FLOAT_MANTISSA_BITS - 1));
67-
assertTrue("" + scaled, scaled < 1 << WANDScorer.FLOAT_MANTISSA_BITS);
69+
assertThat(scaled, greaterThanOrEqualTo(1 << (WANDScorer.FLOAT_MANTISSA_BITS - 1)));
70+
assertThat(scaled, lessThan(1 << WANDScorer.FLOAT_MANTISSA_BITS));
6871
}
6972

7073
public void testScaleMaxScore() {

lucene/core/src/test/org/apache/lucene/search/TestWildcardQuery.java

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
package org.apache.lucene.search;
1818

1919
import static org.hamcrest.Matchers.instanceOf;
20+
import static org.hamcrest.Matchers.not;
2021

2122
import java.io.IOException;
2223
import org.apache.lucene.document.Document;
@@ -77,31 +78,31 @@ public void testTermWithoutWildcard() throws IOException {
7778
new Term("field", "nowildcard"),
7879
Operations.DEFAULT_DETERMINIZE_WORK_LIMIT,
7980
MultiTermQuery.SCORING_BOOLEAN_REWRITE));
80-
assertTrue(q instanceof TermQuery);
81+
assertThat(q, instanceOf(TermQuery.class));
8182

8283
q =
8384
searcher.rewrite(
8485
new WildcardQuery(
8586
new Term("field", "nowildcard"),
8687
Operations.DEFAULT_DETERMINIZE_WORK_LIMIT,
8788
MultiTermQuery.CONSTANT_SCORE_REWRITE));
88-
assertTrue(q instanceof MultiTermQueryConstantScoreWrapper);
89+
assertThat(q, instanceOf(MultiTermQueryConstantScoreWrapper.class));
8990

9091
q =
9192
searcher.rewrite(
9293
new WildcardQuery(
9394
new Term("field", "nowildcard"),
9495
Operations.DEFAULT_DETERMINIZE_WORK_LIMIT,
9596
MultiTermQuery.CONSTANT_SCORE_BLENDED_REWRITE));
96-
assertTrue(q instanceof MultiTermQueryConstantScoreBlendedWrapper);
97+
assertThat(q, instanceOf(MultiTermQueryConstantScoreBlendedWrapper.class));
9798

9899
q =
99100
searcher.rewrite(
100101
new WildcardQuery(
101102
new Term("field", "nowildcard"),
102103
Operations.DEFAULT_DETERMINIZE_WORK_LIMIT,
103104
MultiTermQuery.CONSTANT_SCORE_BOOLEAN_REWRITE));
104-
assertTrue(q instanceof ConstantScoreQuery);
105+
assertThat(q, instanceOf(ConstantScoreQuery.class));
105106
reader.close();
106107
indexStore.close();
107108
}
@@ -119,7 +120,7 @@ public void testEmptyTerm() throws IOException {
119120
MultiTermQuery.SCORING_BOOLEAN_REWRITE);
120121
assertMatches(searcher, wq, 0);
121122
Query q = searcher.rewrite(wq);
122-
assertTrue(q instanceof MatchNoDocsQuery);
123+
assertThat(q, instanceOf(MatchNoDocsQuery.class));
123124
reader.close();
124125
indexStore.close();
125126
}
@@ -139,7 +140,7 @@ public void testPrefixTerm() throws IOException {
139140
wq = new WildcardQuery(new Term("field", "*"));
140141
assertMatches(searcher, wq, 2);
141142
Terms terms = MultiTerms.getTerms(searcher.getIndexReader(), "field");
142-
assertThat(wq.getTermsEnum(terms), instanceOf(AutomatonTermsEnum.class));
143+
assertThat(wq.getTermsEnum(terms), not(instanceOf(AutomatonTermsEnum.class)));
143144
reader.close();
144145
indexStore.close();
145146
}

0 commit comments

Comments
 (0)