Package org.apache.lucene.facet.enhancements.params

Examples of org.apache.lucene.facet.enhancements.params.DefaultEnhancementsIndexingParams

33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
public class DefaultEnhancementsIndexingParamsTest extends LuceneTestCase {

  @Test
  public void testCategoryEnhancements() {
    EnhancementsIndexingParams params =
      new DefaultEnhancementsIndexingParams(
          new CategoryEnhancementDummy1());

    // check retainable properties
    List<Class<? extends CategoryProperty>> retainableProps = params
        .getRetainableProperties();
    assertNull("Unexpected content in retainable list", retainableProps);

    params.addCategoryEnhancements(new CategoryEnhancementDummy2());

    List<CategoryEnhancement> enhancements = params
        .getCategoryEnhancements();

    assertEquals("Wrong number of enhancements", 2, enhancements.size());

    assertTrue("Wrong first enhancement",
        enhancements.get(0) instanceof CategoryEnhancementDummy1);
    assertTrue("Wrong second enhancement",
        enhancements.get(1) instanceof CategoryEnhancementDummy2);

    // re-check retainable properties
    retainableProps = params.getRetainableProperties();
    assertNotNull("Unexpected empty retainable list", retainableProps);
    assertEquals("Unexpected size of retainable list", 1, retainableProps
        .size());
    assertEquals("Wrong property in retainable list", DummyProperty.class,
        retainableProps.get(0));
View Full Code Here
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
  public void testTwoEmptyAndNonEmptyByteArrays() throws Exception {
    Directory indexDir = newDirectory();
    Directory taxoDir = newDirectory();

    EnhancementsIndexingParams indexingParams =
      new DefaultEnhancementsIndexingParams(
          new CategoryEnhancementDummy1(),
          new CategoryEnhancementDummy3());

    // add document with a category containing data for both enhancements
    List<CategoryPath> categoryPaths = new ArrayList<CategoryPath>();
    categoryPaths.add(new CategoryPath("a", "b"));

    RandomIndexWriter indexWriter = new RandomIndexWriter(random, indexDir, newIndexWriterConfig(
        TEST_VERSION_CURRENT, new MockAnalyzer(random, MockTokenizer.WHITESPACE, false)));
    TaxonomyWriter taxo = new LuceneTaxonomyWriter(taxoDir);

    // a category document builder will add the categories to a document
    // once build() is called
    Document doc = new Document();
    indexWriter.addDocument(new EnhancementsDocumentBuilder(taxo,
        indexingParams).setCategoryPaths(categoryPaths).build(doc));

    IndexReader indexReader = indexWriter.getReader();
    indexWriter.close();

    Term term = DrillDown.term(indexingParams, new CategoryPath("a","b"));
    EnhancementsPayloadIterator iterator = new EnhancementsPayloadIterator(
        indexingParams.getCategoryEnhancements(), indexReader, term);

    assertTrue("EnhancementsPayloadIterator failure", iterator.init());
    assertTrue("Missing document 0", iterator.setdoc(0));
    assertNull("Unexpected data for CategoryEnhancementDummy2", iterator
        .getCategoryData(new CategoryEnhancementDummy1()));
View Full Code Here
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
    // add document with a category containing data for both enhancements
    Directory indexDir = newDirectory();
    Directory taxoDir = newDirectory();

    EnhancementsIndexingParams indexingParams =
      new DefaultEnhancementsIndexingParams(
          new CategoryEnhancementDummy2(),
          new CategoryEnhancementDummy3());

    List<CategoryPath> categoryPaths = new ArrayList<CategoryPath>();
    categoryPaths.add(new CategoryPath("a", "b"));

    RandomIndexWriter indexWriter = new RandomIndexWriter(random, indexDir, newIndexWriterConfig(
        TEST_VERSION_CURRENT, new MockAnalyzer(random, MockTokenizer.WHITESPACE, false)));
    TaxonomyWriter taxo = new LuceneTaxonomyWriter(taxoDir);

    // a category document builder will add the categories to a document
    // once build() is called
    Document doc = new Document();
    indexWriter.addDocument(new EnhancementsDocumentBuilder(taxo,
        indexingParams).setCategoryPaths(categoryPaths).build(doc));

    IndexReader indexReader = indexWriter.getReader();
    indexWriter.close();

    Term term = DrillDown.term(indexingParams, new CategoryPath("a","b"));
    EnhancementsPayloadIterator iterator = new EnhancementsPayloadIterator(
        indexingParams.getCategoryEnhancements(), indexReader, term);

    assertTrue("EnhancementsPayloadIterator failure", iterator.init());
    assertTrue("Missing document 0", iterator.setdoc(0));
    byte[] dummy2 = (byte[]) iterator
        .getCategoryData(new CategoryEnhancementDummy2());
View Full Code Here
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
        throw new UnsupportedOperationException();
      }
    }

    final int NUM_CATEGORIES = 10;
    EnhancementsIndexingParams iParams = new DefaultEnhancementsIndexingParams(
        new AssociationEnhancement());

    Directory iDir = newDirectory();
    Directory tDir = newDirectory();
   
    RandomIndexWriter w = new RandomIndexWriter(random, iDir,
        newIndexWriterConfig(TEST_VERSION_CURRENT, new MockAnalyzer(random, MockTokenizer.KEYWORD, false)));
    LuceneTaxonomyWriter taxoW = new LuceneTaxonomyWriter(tDir);
   
    CategoryContainer cc = new CategoryContainer();
    EnhancementsDocumentBuilder builder = new EnhancementsDocumentBuilder(taxoW, iParams);
    for (int i = 1; i <= NUM_CATEGORIES; i++) {
      CategoryAttributeImpl ca = new CategoryAttributeImpl(new CategoryPath(Integer.toString(i)));
      ca.addProperty(new CustomProperty(i));
     
      cc.addCategory(ca);
    }
    builder.setCategories(cc);
    w.addDocument(builder.build(new Document()));
    taxoW.close();
    IndexReader reader = w.getReader();
    w.close();
   
    LuceneTaxonomyReader taxo = new LuceneTaxonomyReader(tDir);
    String field = iParams.getCategoryListParams(new CategoryPath("0")).getTerm().field();
    AssociationsPayloadIterator api = new AssociationsPayloadIterator(reader, field);

    api.setNextDoc(0);

    boolean flag = false;
View Full Code Here
70
71
72
73
74
75
76
77
78
79
80
        new MockAnalyzer(random, MockTokenizer.KEYWORD, false)));
   
    TaxonomyWriter taxoWriter = new LuceneTaxonomyWriter(taxoDir);
   
    EnhancementsDocumentBuilder builder = new EnhancementsDocumentBuilder(
        taxoWriter, new DefaultEnhancementsIndexingParams(
            new AssociationEnhancement()));
   
    // index documents, 50% have only 'b' and all have 'a'
    for (int i = 0; i < 100; i++) {
      Document doc = new Document();
View Full Code Here
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
  public void testTwoEmptyAndNonEmptyByteArrays() throws Exception {
    Directory indexDir = newDirectory();
    Directory taxoDir = newDirectory();

    EnhancementsIndexingParams indexingParams =
      new DefaultEnhancementsIndexingParams(
          new CategoryEnhancementDummy1(),
          new CategoryEnhancementDummy3());

    // add document with a category containing data for both enhancements
    List<CategoryPath> categoryPaths = new ArrayList<CategoryPath>();
    categoryPaths.add(new CategoryPath("a", "b"));

    RandomIndexWriter indexWriter = new RandomIndexWriter(random, indexDir, newIndexWriterConfig(
        TEST_VERSION_CURRENT, new MockAnalyzer(random, MockTokenizer.WHITESPACE, false)));
    TaxonomyWriter taxo = new DirectoryTaxonomyWriter(taxoDir);

    // a category document builder will add the categories to a document
    // once build() is called
    Document doc = new Document();
    indexWriter.addDocument(new EnhancementsDocumentBuilder(taxo,
        indexingParams).setCategoryPaths(categoryPaths).build(doc));

    IndexReader indexReader = indexWriter.getReader();
    indexWriter.close();

    Term term = DrillDown.term(indexingParams, new CategoryPath("a","b"));
    EnhancementsPayloadIterator iterator = new EnhancementsPayloadIterator(
        indexingParams.getCategoryEnhancements(), indexReader, term);

    assertTrue("EnhancementsPayloadIterator failure", iterator.init());
    assertTrue("Missing document 0", iterator.setdoc(0));
    assertNull("Unexpected data for CategoryEnhancementDummy2", iterator
        .getCategoryData(new CategoryEnhancementDummy1()));
View Full Code Here
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
    // add document with a category containing data for both enhancements
    Directory indexDir = newDirectory();
    Directory taxoDir = newDirectory();

    EnhancementsIndexingParams indexingParams =
      new DefaultEnhancementsIndexingParams(
          new CategoryEnhancementDummy2(),
          new CategoryEnhancementDummy3());

    List<CategoryPath> categoryPaths = new ArrayList<CategoryPath>();
    categoryPaths.add(new CategoryPath("a", "b"));

    RandomIndexWriter indexWriter = new RandomIndexWriter(random, indexDir, newIndexWriterConfig(
        TEST_VERSION_CURRENT, new MockAnalyzer(random, MockTokenizer.WHITESPACE, false)));
    TaxonomyWriter taxo = new DirectoryTaxonomyWriter(taxoDir);

    // a category document builder will add the categories to a document
    // once build() is called
    Document doc = new Document();
    indexWriter.addDocument(new EnhancementsDocumentBuilder(taxo,
        indexingParams).setCategoryPaths(categoryPaths).build(doc));

    IndexReader indexReader = indexWriter.getReader();
    indexWriter.close();

    Term term = DrillDown.term(indexingParams, new CategoryPath("a","b"));
    EnhancementsPayloadIterator iterator = new EnhancementsPayloadIterator(
        indexingParams.getCategoryEnhancements(), indexReader, term);

    assertTrue("EnhancementsPayloadIterator failure", iterator.init());
    assertTrue("Missing document 0", iterator.setdoc(0));
    byte[] dummy2 = (byte[]) iterator
        .getCategoryData(new CategoryEnhancementDummy2());
View Full Code Here
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
        throw new UnsupportedOperationException();
      }
    }

    final int NUM_CATEGORIES = 10;
    EnhancementsIndexingParams iParams = new DefaultEnhancementsIndexingParams(
        new AssociationEnhancement());

    Directory iDir = newDirectory();
    Directory tDir = newDirectory();
   
    RandomIndexWriter w = new RandomIndexWriter(random, iDir,
        newIndexWriterConfig(TEST_VERSION_CURRENT, new MockAnalyzer(random, MockTokenizer.KEYWORD, false)));
    DirectoryTaxonomyWriter taxoW = new DirectoryTaxonomyWriter(tDir);
   
    CategoryContainer cc = new CategoryContainer();
    EnhancementsDocumentBuilder builder = new EnhancementsDocumentBuilder(taxoW, iParams);
    for (int i = 1; i <= NUM_CATEGORIES; i++) {
      CategoryAttributeImpl ca = new CategoryAttributeImpl(new CategoryPath(Integer.toString(i)));
      ca.addProperty(new CustomProperty(i));
     
      cc.addCategory(ca);
    }
    builder.setCategories(cc);
    w.addDocument(builder.build(new Document()));
    taxoW.close();
    IndexReader reader = w.getReader();
    w.close();
   
    DirectoryTaxonomyReader taxo = new DirectoryTaxonomyReader(tDir);
    String field = iParams.getCategoryListParams(new CategoryPath("0")).getTerm().field();
    AssociationsPayloadIterator api = new AssociationsPayloadIterator(reader, field);

    api.setNextDoc(0);

    boolean flag = false;
View Full Code Here
70
71
72
73
74
75
76
77
78
79
80
        new MockAnalyzer(random, MockTokenizer.KEYWORD, false)));
   
    TaxonomyWriter taxoWriter = new DirectoryTaxonomyWriter(taxoDir);
   
    EnhancementsDocumentBuilder builder = new EnhancementsDocumentBuilder(
        taxoWriter, new DefaultEnhancementsIndexingParams(
            new AssociationEnhancement()));
   
    // index documents, 50% have only 'b' and all have 'a'
    for (int i = 0; i < 100; i++) {
      Document doc = new Document();
View Full Code Here
TOP

Related Classes of org.apache.lucene.facet.enhancements.params.DefaultEnhancementsIndexingParams

  • org.apache.lucene.facet.enhancements.association.CustomAssociationPropertyTest
  • org.apache.lucene.facet.enhancements.params.DefaultEnhancementsIndexingParamsTest
  • org.apache.lucene.facet.enhancements.TwoEnhancementsTest
  • org.apache.lucene.facet.search.association.AssociationsFacetRequestTest
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.