ads2_2022/code/rust/tests/test_graphs/test_tarjan.rs

103 lines
2.7 KiB
Rust
Raw Normal View History

// ----------------------------------------------------------------
// IMPORTS
// ----------------------------------------------------------------
use rstest::fixture;
use rstest::rstest;
use std::fmt::Display;
use std::hash::Hash;
use ads2::core::utils;
use ads2::graphs::graph;
use ads2::graphs::tarjan;
// ----------------------------------------------------------------
// Fixtures
// ----------------------------------------------------------------
#[fixture]
fn fixture_graph1() -> graph::Graph<i32> {
return graph::Graph::<i32>::new(
vec![1,2,3,4],
vec![(1,2), (2,4), (4,2)]
);
}
#[fixture]
fn fixture_graph2() -> graph::Graph<i32> {
return graph::Graph::<i32>::new(
vec![1,2,3,4,5,6,7],
vec![(1,2), (1,3), (2,3), (3,4), (4,5), (5,2), (5,6), (5,7), (6,7)]
);
}
#[fixture]
fn fixture_graph3() -> graph::Graph<i32> {
return graph::Graph::<i32>::new(
vec![1,2,3,4,5,6,7,8],
vec![
(1,2),
(1,3),
(2,4),
(2,5),
(3,5),
(3,6),
(3,8),
(4,5),
(4,7),
(5,1),
(5,8),
(6,8),
(7,8),
(8,6),
],
);
}
// ----------------------------------------------------------------
// Test Graph
// ----------------------------------------------------------------
#[rstest]
#[case(fixture_graph1(), vec![vec![1], vec![3], vec![2,4]])]
#[case(fixture_graph2(), vec![vec![1], vec![6], vec![7], vec![2,3,4,5]])]
#[case(fixture_graph3(), vec![vec![1,2,3,4,5], vec![7], vec![6,8]])]
fn test_tarjan<T>(#[case] gph: graph::Graph<T>, #[case] expected: Vec<Vec<T>>)
where T: Eq + Hash + Clone + Display
{
let components = tarjan::tarjan_algorithm(&gph);
assert_components_eq(&components, &expected)
}
// ----------------------------------------------------------------
// AUXILIARY METHODS
// ----------------------------------------------------------------
fn check_components_eq<T>(components1: &Vec<Vec<T>>, components2: &Vec<Vec<T>>) -> bool
where T: Eq + Hash + Clone
{
if components1.len() != components2.len() {
return false
}
for component1 in components1 {
let mut found = false;
for component2 in components2 {
if utils::vec_to_set(component1) == utils::vec_to_set(component2) {
found = true;
break;
}
}
if !found {
return false;
}
}
return true;
}
fn assert_components_eq<T>(components1: &Vec<Vec<T>>, components2: &Vec<Vec<T>>)
where T: Eq + Hash + Clone
{
assert!(check_components_eq(components1, components2));
}