X-Git-Url: https://git.mdrn.pl/pylucene.git/blobdiff_plain/a2e61f0c04805cfcb8706176758d1283c7e3a55c..aaeed5504b982cf3545252ab528713250aa33eed:/lucene-java-3.4.0/lucene/contrib/facet/src/test/org/apache/lucene/facet/index/FacetsPayloadProcessorProviderTest.java diff --git a/lucene-java-3.4.0/lucene/contrib/facet/src/test/org/apache/lucene/facet/index/FacetsPayloadProcessorProviderTest.java b/lucene-java-3.4.0/lucene/contrib/facet/src/test/org/apache/lucene/facet/index/FacetsPayloadProcessorProviderTest.java deleted file mode 100644 index 86fef2c..0000000 --- a/lucene-java-3.4.0/lucene/contrib/facet/src/test/org/apache/lucene/facet/index/FacetsPayloadProcessorProviderTest.java +++ /dev/null @@ -1,114 +0,0 @@ -package org.apache.lucene.facet.index; - -import java.io.IOException; -import java.util.ArrayList; -import java.util.List; - -import org.apache.lucene.analysis.MockAnalyzer; -import org.apache.lucene.analysis.MockTokenizer; -import org.apache.lucene.document.Document; -import org.apache.lucene.index.IndexReader; -import org.apache.lucene.index.IndexWriterConfig; -import org.apache.lucene.index.RandomIndexWriter; -import org.apache.lucene.search.IndexSearcher; -import org.apache.lucene.search.MatchAllDocsQuery; -import org.apache.lucene.store.Directory; -import org.junit.Test; - -import org.apache.lucene.util.LuceneTestCase; -import org.apache.lucene.facet.example.merge.TaxonomyMergeUtils; -import org.apache.lucene.facet.search.FacetsCollector; -import org.apache.lucene.facet.search.params.CountFacetRequest; -import org.apache.lucene.facet.search.params.FacetSearchParams; -import org.apache.lucene.facet.search.results.FacetResult; -import org.apache.lucene.facet.search.results.FacetResultNode; -import org.apache.lucene.facet.taxonomy.CategoryPath; -import org.apache.lucene.facet.taxonomy.lucene.LuceneTaxonomyReader; -import org.apache.lucene.facet.taxonomy.lucene.LuceneTaxonomyWriter; - -/** - * Licensed to the Apache Software Foundation (ASF) under one or more - * contributor license agreements. See the NOTICE file distributed with - * this work for additional information regarding copyright ownership. - * The ASF licenses this file to You under the Apache License, Version 2.0 - * (the "License"); you may not use this file except in compliance with - * the License. You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -public class FacetsPayloadProcessorProviderTest extends LuceneTestCase { - - private static final int NUM_DOCS = 100; - - @Test - public void testTaxonomyMergeUtils() throws Exception { - Directory dir = newDirectory(); - Directory taxDir = newDirectory(); - buildIndexWithFacets(dir, taxDir, true); - - Directory dir1 = newDirectory(); - Directory taxDir1 = newDirectory(); - buildIndexWithFacets(dir1, taxDir1, false); - - TaxonomyMergeUtils.merge(dir, taxDir, dir1, taxDir1); - - verifyResults(dir1, taxDir1); - dir1.close(); - taxDir1.close(); - dir.close(); - taxDir.close(); - } - - private void verifyResults(Directory dir, Directory taxDir) throws IOException { - IndexReader reader1 = IndexReader.open(dir); - LuceneTaxonomyReader taxReader = new LuceneTaxonomyReader(taxDir); - IndexSearcher searcher = newSearcher(reader1); - FacetSearchParams fsp = new FacetSearchParams(); - fsp.addFacetRequest(new CountFacetRequest(new CategoryPath("tag"), NUM_DOCS)); - FacetsCollector collector = new FacetsCollector(fsp, reader1, taxReader); - searcher.search(new MatchAllDocsQuery(), collector); - FacetResult result = collector.getFacetResults().get(0); - FacetResultNode node = result.getFacetResultNode(); - for (FacetResultNode facet: node.getSubResults()) { - int weight = (int)facet.getValue(); - int label = Integer.parseInt(facet.getLabel().getComponent(1)); - //System.out.println(label + ": " + weight); - if (VERBOSE) { - System.out.println(label + ": " + weight); - } - assertEquals(NUM_DOCS ,weight); - } - reader1.close(); - taxReader.close(); - } - - private void buildIndexWithFacets(Directory dir, Directory taxDir, boolean asc) throws IOException { - IndexWriterConfig config = newIndexWriterConfig(TEST_VERSION_CURRENT, - new MockAnalyzer(random, MockTokenizer.WHITESPACE, false)); - RandomIndexWriter writer = new RandomIndexWriter(random, dir, config); - - LuceneTaxonomyWriter taxonomyWriter = new LuceneTaxonomyWriter(taxDir); - for (int i = 1; i <= NUM_DOCS; i++) { - Document doc = new Document(); - List categoryPaths = new ArrayList(i + 1); - for (int j = i; j <= NUM_DOCS; j++) { - int facetValue = asc? j: NUM_DOCS - j; - categoryPaths.add(new CategoryPath("tag", Integer.toString(facetValue))); - } - CategoryDocumentBuilder catBuilder = new CategoryDocumentBuilder(taxonomyWriter); - catBuilder.setCategoryPaths(categoryPaths); - catBuilder.build(doc); - writer.addDocument(doc); - } - taxonomyWriter.close(); - writer.close(); - } - -}