aboutsummaryrefslogtreecommitdiffstats
path: root/container-search/src/test/java/com/yahoo/search/query/MatchingTestCase.java
blob: 98d5c3870491603cc2a6189ab59266cc6425a1be (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.query;

import com.yahoo.search.Query;
import org.junit.jupiter.api.Test;

import static org.junit.jupiter.api.Assertions.*;

/**
 * @author baldersheim
 */
public class MatchingTestCase {

    @Test
    void testDefaultsInQuery() {
        Query query = new Query("?query=test");
        assertNull(query.getRanking().getMatching().getTermwiseLimit());
        assertNull(query.getRanking().getMatching().getNumThreadsPerSearch());
        assertNull(query.getRanking().getMatching().getNumSearchPartitions());
        assertNull(query.getRanking().getMatching().getMinHitsPerThread());
        assertNull(query.getRanking().getMatching().getPostFilterThreshold());
        assertNull(query.getRanking().getMatching().getApproximateThreshold());
        assertNull(query.getRanking().getMatching().getTargetHitsMaxAdjustmentFactor());
    }

    @Test
    void testQueryOverrides() {
        Query query = new Query("?query=test" +
                "&ranking.matching.termwiseLimit=0.7" +
                "&ranking.matching.numThreadsPerSearch=17" +
                "&ranking.matching.numSearchPartitions=13" +
                "&ranking.matching.minHitsPerThread=3" +
                "&ranking.matching.postFilterThreshold=0.8" +
                "&ranking.matching.approximateThreshold=0.3" +
                "&ranking.matching.targetHitsMaxAdjustmentFactor=2.5");
        assertEquals(Double.valueOf(0.7), query.getRanking().getMatching().getTermwiseLimit());
        assertEquals(Integer.valueOf(17), query.getRanking().getMatching().getNumThreadsPerSearch());
        assertEquals(Integer.valueOf(13), query.getRanking().getMatching().getNumSearchPartitions());
        assertEquals(Integer.valueOf(3), query.getRanking().getMatching().getMinHitsPerThread());
        assertEquals(Double.valueOf(0.8), query.getRanking().getMatching().getPostFilterThreshold());
        assertEquals(Double.valueOf(0.3), query.getRanking().getMatching().getApproximateThreshold());
        assertEquals(Double.valueOf(2.5), query.getRanking().getMatching().getTargetHitsMaxAdjustmentFactor());

        query.prepare();
        assertEquals("0.7", query.getRanking().getProperties().get("vespa.matching.termwise_limit").get(0));
        assertEquals("17", query.getRanking().getProperties().get("vespa.matching.numthreadspersearch").get(0));
        assertEquals("13", query.getRanking().getProperties().get("vespa.matching.numsearchpartitions").get(0));
        assertEquals("3", query.getRanking().getProperties().get("vespa.matching.minhitsperthread").get(0));
        assertEquals("0.8", query.getRanking().getProperties().get("vespa.matching.global_filter.upper_limit").get(0));
        assertEquals("0.3", query.getRanking().getProperties().get("vespa.matching.global_filter.lower_limit").get(0));
        assertEquals("2.5", query.getRanking().getProperties().get("vespa.matching.nns.target_hits_max_adjustment_factor").get(0));
    }

    @Test
    void testBackwardsCompatibleQueryOverrides() {
        // The lowercase aliases are supported to provide backwards compatibility of the properties that was wrongly named in the first place.
        Query query = new Query("?query=test" +
                "&ranking.matching.termwiselimit=0.7" +
                "&ranking.matching.numthreadspersearch=17" +
                "&ranking.matching.numsearchpartitions=13" +
                "&ranking.matching.minhitsperthread=3");
        assertEquals(Double.valueOf(0.7), query.getRanking().getMatching().getTermwiseLimit());
        assertEquals(Integer.valueOf(17), query.getRanking().getMatching().getNumThreadsPerSearch());
        assertEquals(Integer.valueOf(13), query.getRanking().getMatching().getNumSearchPartitions());
        assertEquals(Integer.valueOf(3), query.getRanking().getMatching().getMinHitsPerThread());
    }

    private void verifyException(String key, String value) {
        try {
            new Query("?query=test&ranking.matching."+key+"="+value);
            fail();
        } catch (IllegalArgumentException e) {
            assertEquals("Could not set 'ranking.matching." + key + "' to '" + value +"'", e.getMessage());
            assertEquals(key + " must be in the range [0.0, 1.0]. It is " + value, e.getCause().getMessage());
        }
    }

    @Test
    void testLimits() {
        verifyException("termwiselimit", "-0.1");
        verifyException("termwiselimit", "1.1");
    }

}