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/search/TestFacetsAccumulatorWithComplement.java?ds=sidebyside diff --git a/lucene-java-3.4.0/lucene/contrib/facet/src/test/org/apache/lucene/facet/search/TestFacetsAccumulatorWithComplement.java b/lucene-java-3.4.0/lucene/contrib/facet/src/test/org/apache/lucene/facet/search/TestFacetsAccumulatorWithComplement.java deleted file mode 100644 index 3609158..0000000 --- a/lucene-java-3.4.0/lucene/contrib/facet/src/test/org/apache/lucene/facet/search/TestFacetsAccumulatorWithComplement.java +++ /dev/null @@ -1,159 +0,0 @@ -package org.apache.lucene.facet.search; - -import java.io.IOException; -import java.util.List; - -import org.apache.lucene.index.IndexReader; -import org.apache.lucene.index.MultiReader; -import org.apache.lucene.index.ParallelReader; -import org.apache.lucene.search.MatchAllDocsQuery; -import org.apache.lucene.search.Query; -import org.junit.After; -import org.junit.Before; -import org.junit.Test; - -import org.apache.lucene.facet.FacetTestBase; -import org.apache.lucene.facet.search.FacetsAccumulator; -import org.apache.lucene.facet.search.ScoredDocIDs; -import org.apache.lucene.facet.search.ScoredDocIdCollector; -import org.apache.lucene.facet.search.StandardFacetsAccumulator; -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; - -/** - * 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. - */ - -/** - * Test that complementsworks as expected. - * We place this test under *.facet.search rather than *.search - * because the test actually does faceted search. - */ -public class TestFacetsAccumulatorWithComplement extends FacetTestBase { - - @Override - @Before - public void setUp() throws Exception { - super.setUp(); - initIndex(); - } - - @Override - @After - public void tearDown() throws Exception { - closeAll(); - super.tearDown(); - } - - /** - * Test that complements does not cause a failure when using a parallel reader - */ - @Test - public void testComplementsWithParallerReader() throws Exception { - IndexReader origReader = indexReader; - ParallelReader pr = new ParallelReader(true); - pr.add(origReader); - indexReader = pr; - try { - doTestComplements(); - } finally { - indexReader = origReader; - } - } - - /** - * Test that complements works with MultiReader - */ - @Test - public void testComplementsWithMultiReader() throws Exception { - final IndexReader origReader = indexReader; - indexReader = new MultiReader(origReader); - try { - doTestComplements(); - } finally { - indexReader = origReader; - } - } - - /** - * Test that score is indeed constant when using a constant score - */ - @Test - public void testComplements() throws Exception { - doTestComplements(); - } - - private void doTestComplements() throws Exception { - Query q = new MatchAllDocsQuery(); //new TermQuery(new Term(TEXT,"white")); - if (VERBOSE) { - System.out.println("Query: "+q); - } - ScoredDocIdCollector dCollector = - ScoredDocIdCollector.create(indexReader.maxDoc(),false); // scoring is disabled - searcher.search(q, dCollector); - - // verify by facet values - List countResWithComplement = findFacets(dCollector.getScoredDocIDs(), true); - List countResNoComplement = findFacets(dCollector.getScoredDocIDs(), false); - - assertEquals("Wrong number of facet count results with complement!",1,countResWithComplement.size()); - assertEquals("Wrong number of facet count results no complement!",1,countResNoComplement.size()); - - FacetResultNode parentResWithComp = countResWithComplement.get(0).getFacetResultNode(); - FacetResultNode parentResNoComp = countResWithComplement.get(0).getFacetResultNode(); - - assertEquals("Wrong number of top count aggregated categories with complement!",3,parentResWithComp.getNumSubResults()); - assertEquals("Wrong number of top count aggregated categories no complement!",3,parentResNoComp.getNumSubResults()); - - } - - @Override - protected FacetSearchParams getFacetedSearchParams() { - FacetSearchParams res = super.getFacetedSearchParams(); - res.addFacetRequest(new CountFacetRequest(new CategoryPath("root","a"), 10)); - return res; - } - - /** compute facets with certain facet requests and docs */ - private List findFacets(ScoredDocIDs sDocids, boolean withComplement) throws IOException { - - FacetsAccumulator fAccumulator = - new StandardFacetsAccumulator(getFacetedSearchParams(), indexReader, taxoReader); - - fAccumulator.setComplementThreshold( - withComplement ? - FacetsAccumulator.FORCE_COMPLEMENT: - FacetsAccumulator.DISABLE_COMPLEMENT); - - List res = fAccumulator.accumulate(sDocids); - - // Results are ready, printing them... - int i = 0; - for (FacetResult facetResult : res) { - if (VERBOSE) { - System.out.println("Res "+(i++)+": "+facetResult); - } - } - - assertEquals(withComplement, ((StandardFacetsAccumulator) fAccumulator).isUsingComplements); - - return res; - } - -} \ No newline at end of file