List of usage examples for org.apache.lucene.index DirectoryReader open
public static DirectoryReader open(final IndexCommit commit) throws IOException
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()))); } }