+++ /dev/null
-package org.apache.lucene.facet.search;
-
-import java.util.Arrays;
-import java.util.List;
-
-import org.apache.lucene.index.IndexReader;
-import org.apache.lucene.search.MatchAllDocsQuery;
-import org.junit.Test;
-
-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.TaxonomyReader;
-
-/**
- * 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 TestTopKResultsHandler extends BaseTestTopK {
-
- private static final CategoryPath[] CATEGORIES = {
- new CategoryPath( "a", "b"),
- new CategoryPath( "a", "b", "1"),
- new CategoryPath( "a", "b", "1"),
- new CategoryPath( "a", "b", "2"),
- new CategoryPath( "a", "b", "2"),
- new CategoryPath( "a", "b", "3"),
- new CategoryPath( "a", "b", "4"),
- new CategoryPath( "a", "c"),
- new CategoryPath( "a", "c"),
- new CategoryPath( "a", "c"),
- new CategoryPath( "a", "c"),
- new CategoryPath( "a", "c"),
- new CategoryPath( "a", "c", "1"),
- };
-
- @Override
- protected String getContent(int doc) {
- return ALPHA;
- }
-
- @Override
- protected int numDocsToIndex() {
- return CATEGORIES.length;
- }
-
- @Override
- protected List<CategoryPath> getCategories(int doc) {
- return Arrays.asList(CATEGORIES[doc]);
- }
-
- /**
- * Strait forward test: Adding specific documents with specific facets and
- * counting them in the most basic form.
- */
- @Test
- public void testSimple() throws Exception {
- for (int partitionSize : partitionSizes) {
- initIndex(partitionSize);
-
- // do different facet counts and compare to control
- FacetSearchParams sParams = getFacetedSearchParams(partitionSize);
-
- sParams.addFacetRequest(new CountFacetRequest(new CategoryPath("a"), 100));
- CountFacetRequest cfra = new CountFacetRequest(new CategoryPath("a"), 100);
- cfra.setDepth(3);
- sParams.addFacetRequest(cfra);
- sParams.addFacetRequest(new CountFacetRequest(new CategoryPath("a", "b"), 100));
- sParams.addFacetRequest(new CountFacetRequest(new CategoryPath("a", "b", "1"), 100));
- sParams.addFacetRequest(new CountFacetRequest(new CategoryPath("a", "c"), 100));
-
- FacetsCollector fc = new FacetsCollector(sParams, indexReader, taxoReader) {
- @Override
- protected FacetsAccumulator initFacetsAccumulator(FacetSearchParams facetSearchParams, IndexReader indexReader, TaxonomyReader taxonomyReader) {
- FacetsAccumulator fa = new StandardFacetsAccumulator(facetSearchParams, indexReader, taxonomyReader);
- fa.setComplementThreshold(FacetsAccumulator.DISABLE_COMPLEMENT);
- return fa;
- }
- };
-
- searcher.search(new MatchAllDocsQuery(), fc);
- long start = System.currentTimeMillis();
- List<FacetResult> facetResults = fc.getFacetResults();
- long end = System.currentTimeMillis();
-
- if (VERBOSE) {
- System.out.println("Time: " + (end - start));
- }
-
- FacetResult fr = facetResults.get(0);
- FacetResultNode parentRes = fr.getFacetResultNode();
- assertEquals(13.0, parentRes.getValue(), Double.MIN_VALUE);
- FacetResultNode[] frn = resultNodesAsArray(parentRes);
- assertEquals(7.0, frn[0].getValue(), Double.MIN_VALUE);
- assertEquals(6.0, frn[1].getValue(), Double.MIN_VALUE);
-
- fr = facetResults.get(1);
- parentRes = fr.getFacetResultNode();
- assertEquals(13.0, parentRes.getValue(), Double.MIN_VALUE);
- frn = resultNodesAsArray(parentRes);
- assertEquals(7.0, frn[0].getValue(), Double.MIN_VALUE);
- assertEquals(6.0, frn[1].getValue(), Double.MIN_VALUE);
- assertEquals(2.0, frn[2].getValue(), Double.MIN_VALUE);
- assertEquals(2.0, frn[3].getValue(), Double.MIN_VALUE);
- assertEquals(1.0, frn[4].getValue(), Double.MIN_VALUE);
- assertEquals(1.0, frn[5].getValue(), Double.MIN_VALUE);
-
- fr = facetResults.get(2);
- parentRes = fr.getFacetResultNode();
- assertEquals(7.0, parentRes.getValue(), Double.MIN_VALUE);
- frn = resultNodesAsArray(parentRes);
- assertEquals(2.0, frn[0].getValue(), Double.MIN_VALUE);
- assertEquals(2.0, frn[1].getValue(), Double.MIN_VALUE);
- assertEquals(1.0, frn[2].getValue(), Double.MIN_VALUE);
- assertEquals(1.0, frn[3].getValue(), Double.MIN_VALUE);
-
- fr = facetResults.get(3);
- parentRes = fr.getFacetResultNode();
- assertEquals(2.0, parentRes.getValue(), Double.MIN_VALUE);
- frn = resultNodesAsArray(parentRes);
- assertEquals(0, frn.length);
-
- fr = facetResults.get(4);
- parentRes = fr.getFacetResultNode();
- assertEquals(6.0, parentRes.getValue(), Double.MIN_VALUE);
- frn = resultNodesAsArray(parentRes);
- assertEquals(1.0, frn[0].getValue(), Double.MIN_VALUE);
- closeAll();
- }
- }
-
- /**
- * Creating an index, matching the results of an top K = Integer.MAX_VALUE and top-1000 requests
- */
- @Test
- public void testGetMaxIntFacets() throws Exception {
- for (int partitionSize : partitionSizes) {
- initIndex(partitionSize);
-
- // do different facet counts and compare to control
- CategoryPath path = new CategoryPath("a", "b");
- FacetSearchParams sParams = getFacetedSearchParams(partitionSize);
- sParams.addFacetRequest(new CountFacetRequest(path, Integer.MAX_VALUE));
-
- FacetsCollector fc = new FacetsCollector(sParams, indexReader, taxoReader) {
- @Override
- protected FacetsAccumulator initFacetsAccumulator(FacetSearchParams facetSearchParams, IndexReader indexReader, TaxonomyReader taxonomyReader) {
- FacetsAccumulator fa = new StandardFacetsAccumulator(facetSearchParams, indexReader, taxonomyReader);
- fa.setComplementThreshold(FacetsAccumulator.DISABLE_COMPLEMENT);
- return fa;
- }
- };
-
- searcher.search(new MatchAllDocsQuery(), fc);
- long start = System.currentTimeMillis();
- List<FacetResult> results = fc.getFacetResults();
- long end = System.currentTimeMillis();
-
- if (VERBOSE) {
- System.out.println("Time: " + (end - start));
- }
-
- assertEquals("Should only be one result as there's only one request", 1, results.size());
- FacetResult res = results.get(0);
- assertEquals(path + " should only have 4 desendants", 4, res.getNumValidDescendants());
-
- // As a control base results, ask for top-1000 results
- FacetSearchParams sParams2 = getFacetedSearchParams(partitionSize);
- sParams2.addFacetRequest(new CountFacetRequest(path, Integer.MAX_VALUE));
-
- FacetsCollector fc2 = new FacetsCollector(sParams2, indexReader, taxoReader) {
- @Override
- protected FacetsAccumulator initFacetsAccumulator(FacetSearchParams facetSearchParams, IndexReader indexReader, TaxonomyReader taxonomyReader) {
- FacetsAccumulator fa = new StandardFacetsAccumulator(facetSearchParams, indexReader, taxonomyReader);
- fa.setComplementThreshold(FacetsAccumulator.DISABLE_COMPLEMENT);
- return fa;
- }
- };
-
- searcher.search(new MatchAllDocsQuery(), fc2);
- List<FacetResult> baseResults = fc2.getFacetResults();
- FacetResult baseRes = baseResults.get(0);
-
- // Removing the first line which holds the REQUEST and this is surly different between the two
- String baseResultString = baseRes.toString();
- baseResultString = baseResultString.substring(baseResultString.indexOf('\n'));
-
- // Removing the first line
- String resultString = res.toString();
- resultString = resultString.substring(resultString.indexOf('\n'));
-
- assertTrue("Results for k=MAX_VALUE do not match the regular results for k=1000!!",
- baseResultString.equals(resultString));
-
- closeAll();
- }
- }
-
- @Test
- public void testSimpleSearchForNonexistentFacet() throws Exception {
- for (int partitionSize : partitionSizes) {
- initIndex(partitionSize);
-
- CategoryPath path = new CategoryPath("Miau Hattulla");
- FacetSearchParams sParams = getFacetedSearchParams(partitionSize);
- sParams.addFacetRequest(new CountFacetRequest(path, 10));
-
- FacetsCollector fc = new FacetsCollector(sParams, indexReader, taxoReader);
-
- searcher.search(new MatchAllDocsQuery(), fc);
-
- long start = System.currentTimeMillis();
- List<FacetResult> facetResults = fc.getFacetResults();
- long end = System.currentTimeMillis();
-
- if (VERBOSE) {
- System.out.println("Time: " + (end - start));
- }
-
- assertEquals("Shouldn't have found anything for a FacetRequest "
- + "of a facet that doesn't exist in the index.", 0, facetResults.size());
-
- closeAll();
- }
- }
-}
fnp / pylucene.git / blobdiff