Search in sources :

Example 1 with FloatColumnReference

use of io.crate.expression.reference.doc.lucene.FloatColumnReference in project crate by crate.

the class FloatColumnReferenceTest method testFieldCacheExpression.

@Test
public void testFieldCacheExpression() throws Exception {
    FloatColumnReference floatColumn = new FloatColumnReference(column);
    floatColumn.startCollect(ctx);
    floatColumn.setNextReader(new ReaderContext(readerContext));
    IndexSearcher searcher = new IndexSearcher(readerContext.reader());
    TopDocs topDocs = searcher.search(new MatchAllDocsQuery(), 10);
    float f = -0.5f;
    for (ScoreDoc doc : topDocs.scoreDocs) {
        floatColumn.setNextDocId(doc.doc);
        assertThat(floatColumn.value(), is(f));
        f++;
    }
}
Also used : IndexSearcher(org.apache.lucene.search.IndexSearcher) TopDocs(org.apache.lucene.search.TopDocs) ReaderContext(io.crate.execution.engine.fetch.ReaderContext) FloatColumnReference(io.crate.expression.reference.doc.lucene.FloatColumnReference) MatchAllDocsQuery(org.apache.lucene.search.MatchAllDocsQuery) ScoreDoc(org.apache.lucene.search.ScoreDoc) Test(org.junit.Test)

Aggregations

ReaderContext (io.crate.execution.engine.fetch.ReaderContext)1 FloatColumnReference (io.crate.expression.reference.doc.lucene.FloatColumnReference)1 IndexSearcher (org.apache.lucene.search.IndexSearcher)1 MatchAllDocsQuery (org.apache.lucene.search.MatchAllDocsQuery)1 ScoreDoc (org.apache.lucene.search.ScoreDoc)1 TopDocs (org.apache.lucene.search.TopDocs)1 Test (org.junit.Test)1