graph-api-test 0.2.1

Test utilities and property-based testing for the graph-api ecosystem
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

use crate::{Edge, Vertex, populate_graph};
use graph_api_lib::{EdgeReference, EdgeSearch, Graph, VertexReference, VertexSearch};

/// Test vertex fold operations
pub fn test_vertices_fold<G>(graph: &mut G)
where
    G: Graph<Vertex = Vertex, Edge = Edge>,
{
    // Populate the graph with test data
    let _refs = populate_graph(graph);

    // Calculate the sum of ages using fold
    let total_age = graph
        .walk()
        .vertices(VertexSearch::scan())
        .fold(0, |total, vertex, _| {
            if let Vertex::Person { age, .. } = *vertex.weight() {
                total + age
            } else {
                total
            }
        });

    // Check that the total age is positive - should include Bryn and Julia
    assert_eq!(total_age, 93);
}

/// Test edge fold operations
pub fn test_edges_fold<G>(graph: &mut G)
where
    G: Graph<Vertex = Vertex, Edge = Edge>,
{
    // Populate the graph with test data
    let _refs = populate_graph(graph);

    // Find the relationship year in the graph
    let knows_year = graph
        .walk()
        .vertices(VertexSearch::scan())
        .edges(EdgeSearch::scan())
        .fold(None, |result, edge, _| {
            if let Edge::Knows { since } = edge.weight() {
                Some(*since)
            } else {
                result
            }
        });

    // There should be at least one relationship with a year
    assert!(knows_year.is_some());
    // The standard test data uses 1999
    assert_eq!(knows_year.unwrap(), 1999);
}