graph_api_test/steps/
take.rs

1use crate::{Edge, Vertex, populate_graph};
2use graph_api_lib::{EdgeSearch, Graph};
3
4pub fn test_vertices_take<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, refs.julia])
13            .take(1)
14            .count(),
15        1
16    );
17}
18
19pub fn test_edges_take<T>(graph: &mut T)
20where
21    T: Graph<Vertex = Vertex, Edge = Edge>,
22{
23    let refs = populate_graph(graph);
24    assert_eq!(
25        graph
26            .walk_mut()
27            .vertices_by_id(vec![refs.bryn])
28            .edges(EdgeSearch::scan().outgoing())
29            .take(1)
30            .count(),
31        1
32    );
33}