graph_api_test/steps/
count.rs

1use crate::{Edge, Vertex, populate_graph};
2use graph_api_lib::{EdgeSearch, Graph};
3
4pub fn test_vertices_count<T>(graph: &mut T)
5where
6    T: Graph<Vertex = Vertex, Edge = Edge>,
7{
8    let refs = populate_graph(graph);
9    assert_eq!(
10        graph
11            .walk_mut()
12            .vertices_by_id(vec![refs.bryn])
13            .edges(EdgeSearch::scan().outgoing())
14            .head()
15            .count(),
16        2
17    );
18}
19
20pub fn test_edges_count<T>(graph: &mut T)
21where
22    T: Graph<Vertex = Vertex, Edge = Edge>,
23{
24    let refs = populate_graph(graph);
25    assert_eq!(
26        graph
27            .walk_mut()
28            .vertices_by_id(vec![refs.bryn])
29            .edges(EdgeSearch::scan().outgoing())
30            .count(),
31        2
32    );
33}