Example usage for org.apache.lucene.index DirectoryReader open

List of usage examples for org.apache.lucene.index DirectoryReader open

Introduction

In this page you can find the example usage for org.apache.lucene.index DirectoryReader open.

Prototype

public static DirectoryReader open(final IndexCommit commit) throws IOException 

Source Link

Document

Expert: returns an IndexReader reading the index in the given IndexCommit .

Usage

From source file:de.unihildesheim.iw.lucene.index.FDRIndexDataProviderTest.java

License:Open Source License

@Test
public void testGetTermFrequency() throws Exception {
    try (TestMemIndex idx = new TestMemIndex(Index.TVECTORS)) {
        final DirectoryReader reader = DirectoryReader.open(idx.dir);
        final FilteredDirectoryReader idxReader = new FilteredDirectoryReader.Builder(reader).build();
        final IndexDataProvider idp = new FDRIndexDataProvider.Builder().indexReader(idxReader).build();

        final long result = idp.getTermFrequency();
        Assert.assertEquals("Term frequency mismatch.", 54L, result);
    }/* w  w w  .ja v  a2 s . c o m*/
}

From source file:de.unihildesheim.iw.lucene.index.FDRIndexDataProviderTest.java

License:Open Source License

@Test
public void testGetTermFrequency_term() throws Exception {
    try (TestMemIndex idx = new TestMemIndex(Index.TVECTORS)) {
        final DirectoryReader reader = DirectoryReader.open(idx.dir);
        final FilteredDirectoryReader idxReader = new FilteredDirectoryReader.Builder(reader).build();
        final IndexDataProvider idp = new FDRIndexDataProvider.Builder().indexReader(idxReader).build();

        Assert.assertEquals("Term frequency mismatch.", 9L, idp.getTermFrequency(new BytesRef("value")));
        Assert.assertEquals("Term frequency mismatch.", 3L, idp.getTermFrequency(new BytesRef("document1")));
        Assert.assertEquals("Term frequency mismatch.", 3L, idp.getTermFrequency(new BytesRef("document2")));
        Assert.assertEquals("Term frequency mismatch.", 3L, idp.getTermFrequency(new BytesRef("document3")));
    }/*w  w  w .ja v  a  2 s  .c o  m*/
}

From source file:de.unihildesheim.iw.lucene.index.FDRIndexDataProviderTest.java

License:Open Source License

@SuppressWarnings("ImplicitNumericConversion")
@Test/*from   w ww. j av a 2s  . c  o m*/
public void testGetDocumentFrequency() throws Exception {
    try (TestMemIndex idx = new TestMemIndex(Index.TVECTORS)) {
        final DirectoryReader reader = DirectoryReader.open(idx.dir);
        final FilteredDirectoryReader idxReader = new FilteredDirectoryReader.Builder(reader).build();
        final IndexDataProvider idp = new FDRIndexDataProvider.Builder().indexReader(idxReader).build();

        Assert.assertEquals("Document frequency mismatch.", 3L,
                idp.getDocumentFrequency(new BytesRef("value")));
        Assert.assertEquals("Document frequency mismatch.", 1L,
                idp.getDocumentFrequency(new BytesRef("document1")));
        Assert.assertEquals("Document frequency mismatch.", 1L,
                idp.getDocumentFrequency(new BytesRef("document2")));
        Assert.assertEquals("Document frequency mismatch.", 1L,
                idp.getDocumentFrequency(new BytesRef("document3")));
    }
}

From source file:de.unihildesheim.iw.lucene.index.FDRIndexDataProviderTest.java

License:Open Source License

@SuppressWarnings("ImplicitNumericConversion")
@Test// w ww  .  j  a  va2  s . c  om
public void testGetDocumentIds() throws Exception {
    try (TestMemIndex idx = new TestMemIndex(Index.TVECTORS)) {
        final DirectoryReader reader = DirectoryReader.open(idx.dir);
        final FilteredDirectoryReader idxReader = new FilteredDirectoryReader.Builder(reader).build();
        final IndexDataProvider idp = new FDRIndexDataProvider.Builder().indexReader(idxReader).build();

        final List<Integer> docIds = idp.getDocumentIds().collect(ArrayList<Integer>::new, List::add,
                List::addAll);

        Assert.assertEquals("Number of docIds differ.", idx.docs, docIds.size());
    }
}

From source file:de.unihildesheim.iw.lucene.index.FDRIndexDataProviderTest.java

License:Open Source License

@SuppressWarnings("ImplicitNumericConversion")
@Test/*from w ww .j a va  2s. co m*/
public void testGetDocumentModel() throws Exception {
    try (TestMemIndex idx = new TestMemIndex(Index.TVECTORS)) {
        final DirectoryReader reader = DirectoryReader.open(idx.dir);
        final FilteredDirectoryReader idxReader = new FilteredDirectoryReader.Builder(reader).build();
        final IndexDataProvider idp = new FDRIndexDataProvider.Builder().indexReader(idxReader).build();

        final DocumentModel dm1 = idp.getDocumentModel(0);

        Assert.assertEquals("DocumentModel tf value differs.", 3L, dm1.tf(new BytesRef("document1")));
        Assert.assertEquals("DocumentModel rel-tf value differs.", 3d / 18d,
                dm1.relTf(new BytesRef("document1")), 0d);
        Assert.assertEquals("DocumentModel id differs.", 0L, dm1.id);
        Assert.assertEquals("DocumentModel term count differs.", 12L, dm1.termCount());
    }
}

From source file:de.unihildesheim.iw.lucene.index.FDRIndexDataProviderTest.java

License:Open Source License

@Test
public void testHasDocument() throws Exception {
    try (TestMemIndex idx = new TestMemIndex(Index.TVECTORS)) {
        final DirectoryReader reader = DirectoryReader.open(idx.dir);
        final FilteredDirectoryReader idxReader = new FilteredDirectoryReader.Builder(reader).build();
        final IndexDataProvider idp = new FDRIndexDataProvider.Builder().indexReader(idxReader).build();

        for (int i = 0; i < idx.docs; i++) {
            Assert.assertTrue("Document missing.", idp.hasDocument(i));
        }//from  w  w w  .ja  v a 2 s.c  om

        Assert.assertFalse("Unknown document found.", idp.hasDocument(idx.docs + 1));
    }
}

From source file:de.unihildesheim.iw.lucene.index.FDRIndexDataProviderTest.java

License:Open Source License

@Test
public void testGetDocumentsTerms_single() throws Exception {
    try (TestMemIndex idx = new TestMemIndex(Index.TVECTORS)) {
        final DirectoryReader reader = DirectoryReader.open(idx.dir);
        final FilteredDirectoryReader idxReader = new FilteredDirectoryReader.Builder(reader).build();
        final IndexDataProvider idp = new FDRIndexDataProvider.Builder().indexReader(idxReader).build();

        Stream<BytesRef> terms;
        terms = idp.getDocumentsTerms(new RoaringDocIdSet.Builder(1).add(0).build());
        Assert.assertEquals("Number of terms mismatch.", 12L, terms.count());

        terms = idp.getDocumentsTerms(new RoaringDocIdSet.Builder(1).add(0).build());
        Assert.assertEquals("Number of terms mismatch.", 1L,
                terms.filter(t -> "document1".equals(t.utf8ToString())).count());
    }//from w w  w.java  2  s.co  m
}

From source file:de.unihildesheim.iw.lucene.index.FDRIndexDataProviderTest.java

License:Open Source License

@Test
public void testGetDocumentsTerms_multi() throws Exception {
    try (TestMemIndex idx = new TestMemIndex(Index.TVECTORS)) {
        final DirectoryReader reader = DirectoryReader.open(idx.dir);
        final FilteredDirectoryReader idxReader = new FilteredDirectoryReader.Builder(reader).build();
        final IndexDataProvider idp = new FDRIndexDataProvider.Builder().indexReader(idxReader).build();

        Stream<BytesRef> terms;
        terms = idp.getDocumentsTerms(new RoaringDocIdSet.Builder(3).add(0).add(2).build());
        Assert.assertEquals("Number of terms mismatch.", 16L, terms.count());

        terms = idp.getDocumentsTerms(new RoaringDocIdSet.Builder(3).add(0).add(2).build());
        Assert.assertEquals("Number of terms mismatch.", 1L,
                terms.filter(t -> "document1".equals(t.utf8ToString())).count());

        terms = idp.getDocumentsTerms(new RoaringDocIdSet.Builder(3).add(0).add(2).build());
        Assert.assertEquals("Number of terms mismatch.", 1L,
                terms.filter(t -> "document3".equals(t.utf8ToString())).count());
    }/*from   w ww. ja  v a2  s.  c  o  m*/
}

From source file:de.unihildesheim.iw.lucene.index.FDRIndexDataProviderTest.java

License:Open Source License

@SuppressWarnings("ImplicitNumericConversion")
@Test/*from   www  .  j  a  v  a  2 s  . co m*/
public void testGetDocumentCount() throws Exception {
    try (TestMemIndex idx = new TestMemIndex(Index.TVECTORS)) {
        final DirectoryReader reader = DirectoryReader.open(idx.dir);
        final FilteredDirectoryReader idxReader = new FilteredDirectoryReader.Builder(reader).build();
        final IndexDataProvider idp = new FDRIndexDataProvider.Builder().indexReader(idxReader).build();

        Assert.assertEquals("Unknown document found.", idx.docs, idp.getDocumentCount());
    }
}

From source file:de.unihildesheim.iw.lucene.index.FDRIndexDataProviderTest.java

License:Open Source License

@SuppressWarnings("ImplicitNumericConversion")
@Test/*from   www  .ja  va2  s . c  o m*/
public void testGetDocumentFields() throws Exception {
    try (TestMemIndex idx = new TestMemIndex(Index.TVECTORS)) {
        final DirectoryReader reader = DirectoryReader.open(idx.dir);
        final FilteredDirectoryReader idxReader = new FilteredDirectoryReader.Builder(reader).build();
        final IndexDataProvider idp = new FDRIndexDataProvider.Builder().indexReader(idxReader).build();

        Assert.assertEquals("Field list mismatch.", idx.flds.size(), idp.getDocumentFields().length);
        Assert.assertTrue("Field list mismatch.", idx.flds.containsAll(Arrays.asList(idp.getDocumentFields())));
    }
}