use crate::{Edge, Vertex, assert_elements_eq, populate_graph};
use graph_api_lib::{EdgeSearch, Graph, VertexSearch};
pub fn test_head<T>(graph: &mut T)
where
T: Graph<Vertex = Vertex, Edge = Edge>,
{
let refs = populate_graph(graph);
let collected = graph
.walk()
.vertices_by_id(vec![refs.bryn])
.edges(EdgeSearch::scan().outgoing())
.head()
.collect::<Vec<_>>();
assert_elements_eq!(graph, collected, vec![refs.graph_api, refs.julia]);
}
pub fn test_tail<T>(graph: &mut T)
where
T: Graph<Vertex = Vertex, Edge = Edge>,
{
let refs = populate_graph(graph);
let collected = graph
.walk()
.vertices_by_id(vec![refs.bryn])
.edges(EdgeSearch::scan().outgoing())
.tail()
.collect::<Vec<_>>();
assert_elements_eq!(graph, collected, vec![refs.bryn]);
}
pub fn test_take<T>(graph: &mut T)
where
T: Graph<Vertex = Vertex, Edge = Edge>,
{
let refs = populate_graph(graph);
let collected = graph
.walk()
.vertices(VertexSearch::scan().with_limit(1))
.collect::<Vec<_>>();
assert_elements_eq!(graph, collected, vec![refs.bryn]);
}