uni-query 1.1.0

OpenCypher query parser, planner, and vectorized executor for Uni
Documentation 
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
95
96
97
98
99

// SPDX-License-Identifier: Apache-2.0
// Copyright 2024-2026 Dragonscale Team

use serde_json::json;
use uni_common::Value;
use uni_query::query::expr_eval::eval_scalar_function;

/// Helper to convert json! value to uni_common::Value
fn v(j: serde_json::Value) -> Value {
    j.into()
}

#[test]
fn test_point_geographic() {
    let arg = v(json!({"latitude": 12.9716, "longitude": 77.5946}));
    let res = eval_scalar_function("POINT", &[arg], None).unwrap();

    assert_eq!(res.get("crs"), Some(&Value::String("WGS84".to_string())));
    assert_eq!(res.get("latitude"), Some(&Value::Float(12.9716)));
    assert_eq!(res.get("longitude"), Some(&Value::Float(77.5946)));
}

#[test]
fn test_point_cartesian() {
    let arg = v(json!({"x": 10.0, "y": 20.0}));
    let res = eval_scalar_function("POINT", &[arg], None).unwrap();

    assert_eq!(
        res.get("crs"),
        Some(&Value::String("Cartesian".to_string()))
    );
    assert_eq!(res.get("x"), Some(&Value::Float(10.0)));
    assert_eq!(res.get("y"), Some(&Value::Float(20.0)));
}

#[test]
fn test_distance_cartesian() {
    let p1 = v(json!({"crs": "Cartesian", "x": 0.0, "y": 0.0}));
    let p2 = v(json!({"crs": "Cartesian", "x": 3.0, "y": 4.0}));

    let res = eval_scalar_function("DISTANCE", &[p1, p2], None).unwrap();
    assert_eq!(res.as_f64().unwrap(), 5.0);
}

#[test]
fn test_distance_geographic() {
    // Bangalore to Mumbai (~840km)
    let p1 = v(json!({"crs": "WGS84", "latitude": 12.9716, "longitude": 77.5946}));
    let p2 = v(json!({"crs": "WGS84", "latitude": 19.0760, "longitude": 72.8777}));

    let res = eval_scalar_function("DISTANCE", &[p1, p2], None).unwrap();
    let distance_km = res.as_f64().unwrap() / 1000.0;

    assert!(distance_km > 800.0 && distance_km < 900.0);
}

#[test]
fn test_within_bbox_cartesian() {
    let point = v(json!({"crs": "Cartesian", "x": 5.0, "y": 5.0}));
    let lower_left = v(json!({"crs": "Cartesian", "x": 0.0, "y": 0.0}));
    let upper_right = v(json!({"crs": "Cartesian", "x": 10.0, "y": 10.0}));

    // Point inside bbox
    let res = eval_scalar_function(
        "POINT.WITHINBBOX",
        &[point.clone(), lower_left.clone(), upper_right.clone()],
        None,
    )
    .unwrap();
    assert!(res.as_bool().unwrap());

    // Point outside bbox
    let outside = v(json!({"crs": "Cartesian", "x": 15.0, "y": 5.0}));
    let res = eval_scalar_function(
        "POINT.WITHINBBOX",
        &[outside, lower_left, upper_right],
        None,
    )
    .unwrap();
    assert!(!res.as_bool().unwrap());
}

#[test]
fn test_within_bbox_geographic() {
    // Point in San Francisco
    let sf = v(json!({"crs": "WGS84", "latitude": 37.7749, "longitude": -122.4194}));
    // Bounding box covering California
    let ll = v(json!({"crs": "WGS84", "latitude": 32.0, "longitude": -125.0}));
    let ur = v(json!({"crs": "WGS84", "latitude": 42.0, "longitude": -114.0}));

    let res =
        eval_scalar_function("POINT.WITHINBBOX", &[sf, ll.clone(), ur.clone()], None).unwrap();
    assert!(res.as_bool().unwrap());

    // Point in New York (outside California bbox)
    let ny = v(json!({"crs": "WGS84", "latitude": 40.7128, "longitude": -74.0060}));
    let res = eval_scalar_function("POINT.WITHINBBOX", &[ny, ll, ur], None).unwrap();
    assert!(!res.as_bool().unwrap());
}