--- /dev/null
+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<FacetResult> countResWithComplement = findFacets(dCollector.getScoredDocIDs(), true);
+ List<FacetResult> 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<FacetResult> findFacets(ScoredDocIDs sDocids, boolean withComplement) throws IOException {
+
+ FacetsAccumulator fAccumulator =
+ new StandardFacetsAccumulator(getFacetedSearchParams(), indexReader, taxoReader);
+
+ fAccumulator.setComplementThreshold(
+ withComplement ?
+ FacetsAccumulator.FORCE_COMPLEMENT:
+ FacetsAccumulator.DISABLE_COMPLEMENT);
+
+ List<FacetResult> 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
fnp / pylucene.git / blobdiff