aboutsummaryrefslogtreecommitdiffstats
path: root/searchlib/src/tests/common/location/location_test.cpp
blob: aac772b7dd09b0ea53497daa795c7ae8def208c8 (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
85
86
87
88
89
90
91
92
93
94
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/vespalib/testkit/test_kit.h>
#include <vespa/searchlib/common/location.h>
#include <vespa/searchlib/common/geo_location_parser.h>
#include <vespa/searchlib/attribute/attributeguard.h>

using search::common::Location;
using search::common::GeoLocationParser;

Location parse(const char *str) {
    GeoLocationParser parser;
    if (!EXPECT_TRUE(parser.parseOldFormat(str))) {
        fprintf(stderr, "  parse error: %s\n", parser.getParseError());
    }
    return Location(parser.getGeoLocation());
}

TEST("require that bounding boxes can be parsed") {
    Location loc = parse("[2,10,20,30,40]");
    EXPECT_EQUAL(false, loc.getRankOnDistance());
    EXPECT_EQUAL(true, loc.getPruneOnDistance());
    EXPECT_EQUAL(0u, loc.x_aspect.multiplier);
    EXPECT_EQUAL(0, loc.point.x);
    EXPECT_EQUAL(0, loc.point.y);
    EXPECT_EQUAL(std::numeric_limits<uint32_t>::max(), loc.radius);
    EXPECT_EQUAL(10, loc.bounding_box.x.lo);
    EXPECT_EQUAL(20, loc.bounding_box.y.lo);
    EXPECT_EQUAL(30, loc.bounding_box.x.hi);
    EXPECT_EQUAL(40, loc.bounding_box.y.hi);
}

TEST("require that circles can be parsed") {
    Location loc = parse("(2,10,20,5,0,0,0)");
    EXPECT_EQUAL(true, loc.getRankOnDistance());
    EXPECT_EQUAL(true, loc.getPruneOnDistance());
    EXPECT_EQUAL(0u, loc.x_aspect.multiplier);
    EXPECT_EQUAL(10, loc.point.x);
    EXPECT_EQUAL(20, loc.point.y);
    EXPECT_EQUAL(5u, loc.radius);
    EXPECT_EQUAL(5, loc.bounding_box.x.lo);
    EXPECT_EQUAL(15, loc.bounding_box.y.lo);
    EXPECT_EQUAL(15, loc.bounding_box.x.hi);
    EXPECT_EQUAL(25, loc.bounding_box.y.hi);    
}

TEST("require that circles can have aspect ratio") {
    Location loc = parse("(2,10,20,5,0,0,0,2147483648)");
    EXPECT_EQUAL(true, loc.getRankOnDistance());
    EXPECT_EQUAL(true, loc.getPruneOnDistance());
    EXPECT_EQUAL(2147483648u, loc.x_aspect.multiplier);
    EXPECT_EQUAL(10, loc.point.x);
    EXPECT_EQUAL(20, loc.point.y);
    EXPECT_EQUAL(5u, loc.radius);
    EXPECT_EQUAL(-1, loc.bounding_box.x.lo);
    EXPECT_EQUAL(15, loc.bounding_box.y.lo);
    EXPECT_EQUAL(21, loc.bounding_box.x.hi);
    EXPECT_EQUAL(25, loc.bounding_box.y.hi);
}

TEST("require that bounding box can be specified after circle") {
    Location loc = parse("(2,10,20,5,0,0,0)[2,10,20,30,40]");
    EXPECT_EQUAL(true, loc.getRankOnDistance());
    EXPECT_EQUAL(true, loc.getPruneOnDistance());
    EXPECT_EQUAL(0u, loc.x_aspect.multiplier);
    EXPECT_EQUAL(10, loc.point.x);
    EXPECT_EQUAL(20, loc.point.y);
    EXPECT_EQUAL(5u, loc.radius);
    EXPECT_EQUAL(10, loc.bounding_box.x.lo);
    EXPECT_EQUAL(20, loc.bounding_box.y.lo);
    EXPECT_EQUAL(15, loc.bounding_box.x.hi);
    EXPECT_EQUAL(25, loc.bounding_box.y.hi);
}

TEST("require that circles can be specified after bounding box") {
    Location loc = parse("[2,10,20,30,40](2,10,20,5,0,0,0)");
    EXPECT_EQUAL(true, loc.getRankOnDistance());
    EXPECT_EQUAL(true, loc.getPruneOnDistance());
    EXPECT_EQUAL(0u, loc.x_aspect.multiplier);
    EXPECT_EQUAL(10, loc.point.x);
    EXPECT_EQUAL(20, loc.point.y);
    EXPECT_EQUAL(5u, loc.radius);
    EXPECT_EQUAL(10, loc.bounding_box.x.lo);
    EXPECT_EQUAL(20, loc.bounding_box.y.lo);
    EXPECT_EQUAL(15, loc.bounding_box.x.hi);
    EXPECT_EQUAL(25, loc.bounding_box.y.hi);    
}

TEST("require that santa search gives non-wrapped bounding box") {
    Location loc = parse("(2,122163600,89998536,290112,4,2000,0,109704)");
    EXPECT_GREATER_EQUAL(loc.bounding_box.x.hi, loc.bounding_box.x.lo);
    EXPECT_GREATER_EQUAL(loc.bounding_box.y.hi, loc.bounding_box.y.lo);
}

TEST_MAIN() { TEST_RUN_ALL(); }