Refactor algorithm tests to minimize macro code, and move into tests/

The tests only declare a list of tests in the macro, no longer the test functions themselves. This allows better tooling support (e.g. IDE and rustfmt).
This commit is contained in:
2026-08-31 19:44:55 +02:00
parent 6d3a206995
commit df5170f2f5
7 changed files with 1032 additions and 915 deletions
+300 -4
View File
@@ -1,11 +1,307 @@
mod append_graph_tests {
use std::fmt::Debug;
use grapherity::algorithms;
use grapherity::maps::ElementMap;
use grapherity::testing::fixtures::MakeTestGraph;
use grapherity::traits::GraphTopology;
macro_rules! bfs_tests {
($T:ty) => {
bfs_tests!(@wrap $T,
bfs_single_vertex,
bfs_disconnected,
bfs,
bfs_distances_single_vertex,
bfs_distances_disconnected,
bfs_distances,
bfs_find_source,
bfs_find_disconnected,
bfs_find,
bfs_find_where_source_matches,
bfs_find_where_disconnected,
bfs_find_where_no_match,
bfs_find_where_nearest,
);
};
(@wrap $T:ty, $($name:ident),* $(,)?) => {
$(
#[test]
fn $name() {
super::$name::<$T>();
}
)*
};
}
mod append_graph {
use grapherity::models::AppendGraph;
grapherity::bfs_tests!(AppendGraph);
bfs_tests!(AppendGraph);
}
mod graph_tests {
mod graph {
use grapherity::models::Graph;
grapherity::bfs_tests!(Graph);
bfs_tests!(Graph);
}
fn bfs_single_vertex<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, v) = G::single_vertex();
let result = algorithms::bfs(&graph, v);
assert_eq!(
result.distances[v],
Some(0),
"unexpected distance of source vertex"
);
assert_eq!(
result.predecessors[v], None,
"unexpected predecessor of source vertex"
);
}
fn bfs_disconnected<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices) = G::disconnected();
let result = algorithms::bfs(&graph, vertices[0]);
assert_eq!(
result.distances[vertices[0]],
Some(0),
"unexpected distance of source vertex"
);
assert_eq!(
result.predecessors[vertices[0]], None,
"unexpected predecessor of source vertex"
);
for &v in &vertices[1..3] {
assert_eq!(
result.distances[v], None,
"disconnected vertex {v:?} should have no distance"
);
assert_eq!(
result.predecessors[v], None,
"disconnected vertex {v:?} should have no predecessor"
);
}
}
fn bfs<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
let result = algorithms::bfs(&graph, vertices[0]);
assert_bfs_distances::<G>(&result.distances, &vertices);
assert_bfs_predecessors::<G>(&result.predecessors, &vertices);
}
fn bfs_distances_single_vertex<G: MakeTestGraph>() {
let (graph, v) = G::single_vertex();
let distances = algorithms::bfs_distances(&graph, v);
assert_eq!(
distances[v],
Some(0),
"unexpected distance of source vertex"
);
}
fn bfs_distances_disconnected<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices) = G::disconnected();
let distances = algorithms::bfs_distances(&graph, vertices[0]);
assert_eq!(
distances[vertices[0]],
Some(0),
"unexpected distance of source vertex"
);
for &v in &vertices[1..3] {
assert_eq!(
distances[v], None,
"disconnected vertex {v:?} should have no distance"
);
}
}
fn bfs_distances<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
let distances = algorithms::bfs_distances(&graph, vertices[0]);
assert_bfs_distances::<G>(&distances, &vertices);
}
fn bfs_find_source<G: MakeTestGraph>() {
let (graph, v) = G::single_vertex();
assert_eq!(
algorithms::bfs_find(&graph, v, v),
Some(0),
"source should be found at distance 0"
);
}
fn bfs_find_disconnected<G: MakeTestGraph>() {
let (graph, vertices) = G::disconnected();
assert_eq!(
algorithms::bfs_find(&graph, vertices[0], vertices[1]),
None,
"disconnected target should not be found"
);
}
fn bfs_find<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
let expected_distances = [
Some(0),
Some(1),
Some(2),
Some(2),
Some(2),
Some(3),
Some(3),
Some(3),
Some(3),
Some(4),
];
for i in 0..10 {
assert_eq!(
algorithms::bfs_find(&graph, vertices[0], vertices[i]),
expected_distances[i],
"unexpected distance from {:?} to {:?}",
vertices[0],
vertices[i]
);
}
}
fn bfs_find_where_source_matches<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, v) = G::single_vertex();
assert_eq!(
algorithms::bfs_find_where(&graph, v, |u| u == v),
Some((v, 0)),
"source should match with distance 0"
);
}
fn bfs_find_where_disconnected<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices) = G::disconnected();
assert_eq!(
algorithms::bfs_find_where(&graph, vertices[0], |v| v == vertices[1]),
None,
"disconnected vertex {:?} should not be found",
vertices[1]
);
}
fn bfs_find_where_no_match<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
assert_eq!(
algorithms::bfs_find_where(&graph, vertices[0], |_| false),
None,
"no vertex should match an always-false predicate"
);
}
fn bfs_find_where_nearest<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
// vertices[5], vertices[6], vertices[7], vertices[8] are all at distance 3 from vertices[0].
// vertices[9] is at distance 4. Predicate matches vertices[7], vertices[8], vertices[9].
// BFS must return one of the distance-3 ones, not vertices[9].
let result = algorithms::bfs_find_where(&graph, vertices[0], |v| {
v == vertices[7] || v == vertices[8] || v == vertices[9]
});
assert!(result.is_some(), "expected a match");
let (found, distance) = result.unwrap();
assert_eq!(
distance, 3,
"unexpected distance to nearest matching vertex {found:?}",
);
assert!(
found == vertices[7] || found == vertices[8],
"unexpected nearest match vertex {found:?}, should be {:?} or {:?}",
vertices[7],
vertices[8]
);
}
fn assert_bfs_distances<G: GraphTopology>(
distances: &ElementMap<G::Vertex, Option<u32>>,
vertices: &[G::Vertex],
) where
G::Vertex: Debug,
{
let expected = [
Some(0),
Some(1),
Some(2),
Some(2),
Some(2),
Some(3),
Some(3),
Some(3),
Some(3),
Some(4),
];
for i in 0..10 {
assert_eq!(
distances[vertices[i]], expected[i],
"unexpected distance from {:?} to {:?}",
vertices[0], vertices[i]
);
}
}
fn assert_bfs_predecessors<G: GraphTopology>(
predecessors: &ElementMap<G::Vertex, Option<G::Vertex>>,
vertices: &[G::Vertex],
) where
G::Vertex: Debug,
{
assert_eq!(
predecessors[vertices[0]], None,
"source should have no predecessor"
);
// Each non-source vertex's predecessor must be adjacent and at distance one less.
let expected_predecessors = [
vec![None],
vec![Some(vertices[0])],
vec![Some(vertices[1])],
vec![Some(vertices[1])],
vec![Some(vertices[1])],
vec![Some(vertices[2])],
vec![Some(vertices[2]), Some(vertices[3])],
vec![Some(vertices[4])],
vec![Some(vertices[4])],
vec![Some(vertices[5]), Some(vertices[6]), Some(vertices[7])],
];
for i in 1..10 {
assert!(
expected_predecessors[i].contains(&predecessors[vertices[i]]),
"unexpected predecessor {:?} of {:?}",
predecessors[vertices[i]],
vertices[i]
);
}
}
+391 -4
View File
@@ -1,11 +1,398 @@
mod append_graph_tests {
use std::fmt::Debug;
use grapherity::algorithms;
use grapherity::maps::ElementMap;
use grapherity::testing::fixtures::MakeTestGraph;
use grapherity::traits::GraphTopology;
macro_rules! dfs_tests {
($T:ty) => {
dfs_tests!(@wrap $T,
dfs_single_vertex,
dfs_disconnected,
dfs,
dfs_visited_single_vertex,
dfs_visited_disconnected,
dfs_visited,
dfs_find_source,
dfs_find_disconnected,
dfs_find,
dfs_find_where_source_matches,
dfs_find_where_disconnected,
dfs_find_where_no_match,
dfs_find_where_adjacent,
dfs_find_where,
dfs_find_path_source_equals_target,
dfs_find_path_disconnected,
dfs_find_path_adjacent,
dfs_find_path,
dfs_find_path_where_source_matches,
dfs_find_path_where_disconnected,
dfs_find_path_where_no_match,
dfs_find_path_where,
);
};
(@wrap $T:ty, $($name:ident),* $(,)?) => {
$(
#[test]
fn $name() {
super::$name::<$T>();
}
)*
};
}
mod append_graph {
use grapherity::models::AppendGraph;
grapherity::dfs_tests!(AppendGraph);
dfs_tests!(AppendGraph);
}
mod graph_tests {
mod graph {
use grapherity::models::Graph;
grapherity::dfs_tests!(Graph);
dfs_tests!(Graph);
}
fn dfs_single_vertex<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, v) = G::single_vertex();
let result = algorithms::dfs(&graph, v);
assert!(result.visited[v], "source vertex should be visited");
assert_eq!(
result.predecessors[v], None,
"unexpected predecessor of source vertex"
);
}
fn dfs_disconnected<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices) = G::disconnected();
let result = algorithms::dfs(&graph, vertices[0]);
assert!(
result.visited[vertices[0]],
"source vertex should be visited"
);
assert_eq!(
result.predecessors[vertices[0]], None,
"unexpected predecessor of source vertex"
);
for &v in &vertices[1..3] {
assert!(
!result.visited[v],
"disconnected vertex {v:?} should not be visited"
);
assert_eq!(
result.predecessors[v], None,
"disconnected vertex {v:?} should have no predecessor"
);
}
}
fn dfs<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
let result = algorithms::dfs(&graph, vertices[0]);
assert_dfs_visited::<G>(&result.visited, &vertices);
assert_dfs_predecessors(&graph, &result.visited, &result.predecessors, &vertices);
}
fn dfs_visited_single_vertex<G: MakeTestGraph>() {
let (graph, v) = G::single_vertex();
let visited = algorithms::dfs_visited(&graph, v);
assert!(visited[v], "source vertex should be visited");
}
fn dfs_visited_disconnected<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices) = G::disconnected();
let visited = algorithms::dfs_visited(&graph, vertices[0]);
assert!(visited[vertices[0]], "source vertex should be visited");
for &v in &vertices[1..3] {
assert!(
!visited[v],
"disconnected vertex {v:?} should not be visited"
);
}
}
fn dfs_visited<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
let visited = algorithms::dfs_visited(&graph, vertices[0]);
assert_dfs_visited::<G>(&visited, &vertices);
}
fn dfs_find_source<G: MakeTestGraph>() {
let (graph, v) = G::single_vertex();
assert!(
algorithms::dfs_find(&graph, v, v),
"source should find itself"
);
}
fn dfs_find_disconnected<G: MakeTestGraph>() {
let (graph, vertices) = G::disconnected();
assert!(
!algorithms::dfs_find(&graph, vertices[0], vertices[1]),
"disconnected target should not be found"
);
}
fn dfs_find<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
for i in 0..10 {
assert!(
algorithms::dfs_find(&graph, vertices[0], vertices[i]),
"vertex {:?} should be reachable from {:?}",
vertices[i],
vertices[0]
);
}
}
fn dfs_find_where_source_matches<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, v) = G::single_vertex();
assert_eq!(
algorithms::dfs_find_where(&graph, v, |u| u == v),
Some(v),
"source should find itself"
);
}
fn dfs_find_where_disconnected<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices) = G::disconnected();
assert_eq!(
algorithms::dfs_find_where(&graph, vertices[0], |v| v == vertices[1]),
None,
"disconnected vertex {:?} should not be found",
vertices[1]
);
}
fn dfs_find_where_no_match<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
assert_eq!(
algorithms::dfs_find_where(&graph, vertices[0], |_| false),
None,
"no vertex should match an always-false predicate"
);
}
fn dfs_find_where_adjacent<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
assert_eq!(
algorithms::dfs_find_where(&graph, vertices[0], |v| v == vertices[1]),
Some(vertices[1]),
"expected to find adjacent vertex"
);
}
fn dfs_find_where<G: MakeTestGraph>()
where
G::Vertex: Debug,
{
let (graph, vertices, _, _) = G::standard();
assert_eq!(
algorithms::dfs_find_where(&graph, vertices[0], |v| v == vertices[9]),
Some(vertices[9]),
"expected to find connected vertex"
);
}
fn dfs_find_path_source_equals_target<G: MakeTestGraph>()
where
G::Edge: Debug,
{
let (graph, v) = G::single_vertex();
assert_eq!(
algorithms::dfs_find_path(&graph, v, v),
Some(vec![]),
"path from source to itself should be empty"
);
}
fn dfs_find_path_disconnected<G: MakeTestGraph>()
where
G::Edge: Debug,
{
let (graph, vertices) = G::disconnected();
assert_eq!(
algorithms::dfs_find_path(&graph, vertices[0], vertices[1]),
None,
"no path should exist to disconnected vertex"
);
}
fn dfs_find_path_adjacent<G: MakeTestGraph>()
where
G::Edge: Debug,
{
let (graph, vertices, e) = G::single_edge();
let path = algorithms::dfs_find_path(&graph, vertices[0], vertices[1])
.expect("path should exist between adjacent vertices");
assert_eq!(
path.len(),
1,
"unexpected path length between adjacent vertices"
);
assert_eq!(path[0], e, "path should use the connecting edge");
}
fn dfs_find_path<G: MakeTestGraph>()
where
G::Vertex: Debug,
G::Edge: Debug,
{
let (graph, vertices, _, _) = G::standard();
let path = algorithms::dfs_find_path(&graph, vertices[0], vertices[9]).expect(&format!(
"path should exist between connected vertices {:?} and {:?}",
vertices[0], vertices[9]
));
assert_valid_path(&graph, &path, vertices[0], vertices[9]);
}
fn dfs_find_path_where_source_matches<G: MakeTestGraph>()
where
G::Edge: Debug,
{
let (graph, v) = G::single_vertex();
assert_eq!(
algorithms::dfs_find_path_where(&graph, v, |u| u == v),
Some(vec![]),
"path from source to itself should be empty"
);
}
fn dfs_find_path_where_disconnected<G: MakeTestGraph>()
where
G::Edge: Debug,
{
let (graph, vertices) = G::disconnected();
assert_eq!(
algorithms::dfs_find_path_where(&graph, vertices[0], |v| v == vertices[1]),
None,
"no path should exist to disconnected vertex"
);
}
fn dfs_find_path_where_no_match<G: MakeTestGraph>()
where
G::Edge: Debug,
{
let (graph, vertices, _, _) = G::standard();
assert_eq!(
algorithms::dfs_find_path_where(&graph, vertices[0], |_| false),
None,
"no path should exist when predicate never matches"
);
}
fn dfs_find_path_where<G: MakeTestGraph>()
where
G::Vertex: Debug,
G::Edge: Debug,
{
let (graph, vertices, _, _) = G::standard();
let path = algorithms::dfs_find_path_where(&graph, vertices[0], |v| v == vertices[9]).expect(
&format!(
"path should exist between connected vertices {:?} and {:?}",
vertices[0], vertices[9]
),
);
assert_valid_path(&graph, &path, vertices[0], vertices[9]);
}
fn assert_dfs_visited<G: GraphTopology>(
visited: &ElementMap<G::Vertex, bool>,
vertices: &[G::Vertex],
) where
G::Vertex: Debug,
{
for i in 0..10 {
assert!(
visited[vertices[i]],
"vertex {:?} should be visited",
vertices[i]
);
}
}
fn assert_dfs_predecessors<G: GraphTopology>(
graph: &G,
visited: &ElementMap<G::Vertex, bool>,
predecessors: &ElementMap<G::Vertex, Option<G::Vertex>>,
vertices: &[G::Vertex],
) where
G::Vertex: Debug,
{
assert_eq!(
predecessors[vertices[0]], None,
"source should have no predecessor"
);
for i in 1..10 {
let v = vertices[i];
let p = predecessors[v].expect(&format!("vertex {v:?} should have a predecessor"));
assert!(
visited[p],
"predecessor {p:?} of vertex {v:?} should be visited"
);
assert!(
graph.are_adjacent(v, p),
"predecessor {p:?} of vertex {v:?} should be adjacent"
);
}
}
fn assert_valid_path<G: GraphTopology>(
graph: &G,
path: &[G::Edge],
source: G::Vertex,
target: G::Vertex,
) where
G::Vertex: Debug,
G::Edge: Debug,
{
assert!(!path.is_empty(), "path should be non-empty");
// Walks the path: tracks current vertex, confirm each edge is incident to it.
let mut current = source;
for (i, &e) in path.iter().enumerate() {
let (v1, v2) = graph.incident_vertices(e);
assert_ne!(v1, v2, "path should not contain loop edge {e:?}");
assert!(
v1 == current || v2 == current,
"path edge {e:?} (idx {i}, {v1:?} to {v2:?}) not incident to vertex {current:?}"
);
current = if v1 == current { v2 } else { v1 };
}
assert_eq!(
current, target,
"path should end at target {target:?}, but ended at {current:?}"
);
}
+341 -4
View File
@@ -1,11 +1,348 @@
mod append_graph_tests {
use std::fmt::Debug;
use std::hash::Hash;
use grapherity::algorithms;
use grapherity::algorithms::DijkstraResult;
use grapherity::maps::ElementMap;
use grapherity::testing::fixtures::MakeTestGraph;
use grapherity::traits::{GraphTopology, Incidence};
macro_rules! dijkstra_tests {
($T:ty) => {
dijkstra_tests!(@wrap $T,
dijkstra_single_vertex,
dijkstra_disconnected,
dijkstra_zero_weight_loop,
dijkstra,
dijkstra_distances,
dijkstra_unweighted_single_vertex,
dijkstra_unweighted_disconnected,
dijkstra_unweighted,
dijkstra_distances_unweighted_single_vertex,
dijkstra_distances_unweighted_disconnected,
dijkstra_distances_unweighted,
);
};
(@wrap $T:ty, $($name:ident),* $(,)?) => {
$(
#[test]
fn $name() {
super::$name::<$T>();
}
)*
};
}
mod append_graph {
use grapherity::models::AppendGraph;
grapherity::dijkstra_tests!(AppendGraph);
dijkstra_tests!(AppendGraph);
}
mod graph_tests {
mod graph {
use grapherity::models::Graph;
grapherity::dijkstra_tests!(Graph);
dijkstra_tests!(Graph);
}
fn dijkstra_single_vertex<G: MakeTestGraph>()
where
G::Vertex: Debug + Hash,
{
let (graph, v) = G::single_vertex();
let result = algorithms::dijkstra(&graph, v, |_| panic!("unexpected call of weight functor"));
assert_single_vertex::<G>(&result, v);
}
fn dijkstra_disconnected<G: MakeTestGraph>()
where
G::Vertex: Debug + Hash,
{
let (graph, vertices) = G::disconnected();
let result = algorithms::dijkstra(&graph, vertices[0], |_| {
panic!("unexpected call of weight functor")
});
assert_single_vertex::<G>(&result, vertices[0]);
assert_disconnected::<G>(&result, &vertices[1..3]);
}
fn dijkstra_zero_weight_loop<G: MakeTestGraph>()
where
G::Vertex: Debug + Hash,
{
let (graph, vertices, _, weights) = make_two_edge_path_with_zero_weight_loops::<G>();
let result = algorithms::dijkstra(&graph, vertices[0], |e| weights[e]);
assert_single_vertex::<G>(&result, vertices[0]);
for i in 1..3 {
assert_eq!(
result.predecessors[vertices[i]],
Some(vertices[i - 1]),
"unexpected predecessor of vertex {:?}",
vertices[i]
);
assert_eq!(
result.distances[vertices[i]],
Some(i.try_into().unwrap()),
"unexpected distance of vertex {:?}",
vertices[i]
);
}
}
fn dijkstra<G: MakeTestGraph>()
where
G::Vertex: Debug + Hash,
{
let (graph, vertices, _, _, weights) = make_test_graph_weighted::<G>();
let result = algorithms::dijkstra(&graph, vertices[0], |e| weights[e]);
assert_test_graph::<G>(&result, &vertices);
}
fn dijkstra_distances<G: MakeTestGraph>()
where
G::Vertex: Debug + Hash,
{
let (graph, vertices, _, _, weights) = make_test_graph_weighted::<G>();
let distances = algorithms::dijkstra_distances(&graph, vertices[0], |e| weights[e]);
assert_distances_test_graph::<G>(&distances, &vertices);
}
fn dijkstra_unweighted_single_vertex<G: MakeTestGraph>()
where
G::Vertex: Debug + Hash,
{
let (graph, v) = G::single_vertex();
let result = algorithms::dijkstra_unweighted(&graph, v);
assert_single_vertex::<G>(&result, v)
}
fn dijkstra_unweighted_disconnected<G: MakeTestGraph>()
where
G::Vertex: Debug + Hash,
{
let (graph, vertices) = G::disconnected();
let result = algorithms::dijkstra_unweighted(&graph, vertices[0]);
assert_single_vertex::<G>(&result, vertices[0]);
assert_disconnected::<G>(&result, &vertices[1..3]);
}
fn dijkstra_unweighted<G: MakeTestGraph>()
where
G::Vertex: Debug + Hash,
{
let (graph, vertices, _, _) = G::standard();
let result = algorithms::dijkstra_unweighted(&graph, vertices[0]);
assert_unweighted_test_graph::<G>(&result, &vertices);
}
fn dijkstra_distances_unweighted_single_vertex<G: MakeTestGraph>()
where
G::Vertex: Hash,
{
let (graph, v) = G::single_vertex();
let distances = algorithms::dijkstra_distances_unweighted(&graph, v);
assert_distances_single_vertex::<G>(&distances, v);
}
fn dijkstra_distances_unweighted_disconnected<G: MakeTestGraph>()
where
G::Vertex: Debug + Hash,
{
let (graph, vertices) = G::disconnected();
let distances = algorithms::dijkstra_distances_unweighted(&graph, vertices[0]);
assert_distances_single_vertex::<G>(&distances, vertices[0]);
assert_distances_disconnected::<G>(&distances, &vertices[1..3]);
}
fn dijkstra_distances_unweighted<G: MakeTestGraph>()
where
G::Vertex: Debug + Hash,
{
let (graph, vertices, _, _) = G::standard();
let distances = algorithms::dijkstra_distances_unweighted(&graph, vertices[0]);
assert_distances_unweighted_test_graph::<G>(&distances, &vertices);
}
fn assert_single_vertex<G: GraphTopology>(result: &DijkstraResult<G::Vertex>, v: G::Vertex)
where
G::Vertex: Debug,
{
assert_distances_single_vertex::<G>(&result.distances, v);
assert_eq!(
result.predecessors[v], None,
"unexpected predecessor of source vertex",
);
}
fn assert_distances_single_vertex<G: GraphTopology>(
distances: &ElementMap<G::Vertex, Option<u32>>,
v: G::Vertex,
) {
assert_eq!(
distances[v],
Some(0),
"unexpected distance of source vertex"
);
}
fn assert_disconnected<G: GraphTopology>(result: &DijkstraResult<G::Vertex>, vertices: &[G::Vertex])
where
G::Vertex: Debug,
{
assert_distances_disconnected::<G>(&result.distances, &vertices);
for &v in vertices {
assert_eq!(
result.predecessors[v], None,
"unexpected predecessor of disconnected vertex {v:?}",
);
}
}
fn assert_distances_disconnected<G: GraphTopology>(
distances: &ElementMap<G::Vertex, Option<u32>>,
vertices: &[G::Vertex],
) where
G::Vertex: Debug,
{
for &v in vertices {
assert_eq!(
distances[v], None,
"unexpected distance of disconnected vertex {v:?}",
);
}
}
fn assert_test_graph<G: GraphTopology>(result: &DijkstraResult<G::Vertex>, vertices: &[G::Vertex])
where
G::Vertex: Debug,
{
assert_distances_test_graph::<G>(&result.distances, &vertices);
let expected_predecessors_from_v0 = [
vec![None],
vec![Some(vertices[0])],
vec![Some(vertices[4])],
vec![Some(vertices[1])],
vec![Some(vertices[7])],
vec![Some(vertices[9])],
vec![Some(vertices[3])],
vec![Some(vertices[9])],
vec![Some(vertices[7])],
vec![Some(vertices[6])],
];
for i in 0..10 {
assert!(
expected_predecessors_from_v0[i].contains(&result.predecessors[vertices[i]]),
"unexpected predecessor {:?} of {:?}",
result.predecessors[vertices[i]],
vertices[i]
);
}
}
fn assert_distances_test_graph<G: GraphTopology>(
distances: &ElementMap<G::Vertex, Option<u32>>,
vertices: &[G::Vertex],
) where
G::Vertex: Debug,
{
let expected_distances_from_v0 = [
Some(0),
Some(1),
Some(65),
Some(4),
Some(58),
Some(43),
Some(14),
Some(46),
Some(145),
Some(29),
];
for i in 0..10 {
assert_eq!(
distances[vertices[i]], expected_distances_from_v0[i],
"unexpected distance from {:?} to {:?}",
vertices[0], vertices[i]
);
}
}
fn assert_unweighted_test_graph<G: GraphTopology>(
result: &DijkstraResult<G::Vertex>,
vertices: &[G::Vertex],
) where
G::Vertex: Debug,
{
assert_distances_unweighted_test_graph::<G>(&result.distances, &vertices);
let expected_predecessors_from_v0 = [
vec![None],
vec![Some(vertices[0])],
vec![Some(vertices[1])],
vec![Some(vertices[1])],
vec![Some(vertices[1])],
vec![Some(vertices[2])],
vec![Some(vertices[2]), Some(vertices[3])],
vec![Some(vertices[4])],
vec![Some(vertices[4])],
vec![Some(vertices[5]), Some(vertices[6]), Some(vertices[7])],
];
for i in 0..10 {
assert!(
expected_predecessors_from_v0[i].contains(&result.predecessors[vertices[i]]),
"unexpected predecessor {:?} of {:?}",
result.predecessors[vertices[i]],
vertices[i]
);
}
}
fn assert_distances_unweighted_test_graph<G: GraphTopology>(
distances: &ElementMap<G::Vertex, Option<u32>>,
vertices: &[G::Vertex],
) where
G::Vertex: Debug,
{
let expected_distances_from_v0 = [
Some(0),
Some(1),
Some(2),
Some(2),
Some(2),
Some(3),
Some(3),
Some(3),
Some(3),
Some(4),
];
for i in 0..10 {
assert_eq!(
distances[vertices[i]], expected_distances_from_v0[i],
"unexpected distance from {:?} to {:?}",
vertices[0], vertices[i]
);
}
}
fn make_test_graph_weighted<G: MakeTestGraph>() -> (
G,
[G::Vertex; 10],
[(G::Edge, G::Vertex, G::Vertex); 18],
[Vec<Incidence<G::Vertex, G::Edge>>; 10],
ElementMap<G::Edge, u32>,
) {
let (graph, vertices, edges, incidences) = G::standard();
let mut weights = graph.edge_map(99);
for i in [1, 3, 7, 10, 12, 14, 15, 17] {
weights[edges[i].0] = i.try_into().unwrap();
}
(graph, vertices, edges, incidences, weights)
}
fn make_two_edge_path_with_zero_weight_loops<G: MakeTestGraph>()
-> (G, [G::Vertex; 3], [G::Edge; 4], ElementMap<G::Edge, u32>) {
let (graph, vertices, edges) = G::two_edge_path_with_loops();
let mut weights = graph.edge_map(1);
weights[edges[2]] = 0;
weights[edges[3]] = 0;
(graph, vertices, edges, weights)
}