Skip to content

Commit 370d958

Browse files
committed
Replaced assert_eq with methods in traits::graph
1 parent bfc3fec commit 370d958

File tree

4 files changed

+18
-11
lines changed

4 files changed

+18
-11
lines changed

webgraph/src/graphs/arc_list_graph.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,13 +236,13 @@ mod test {
236236
let coo = ArcListGraph::new_labeled(g.num_nodes(), arcs.iter().copied());
237237
let g2 = LabeledBTreeGraph::<_>::from_lender(coo.iter());
238238

239-
assert_eq!(g, g2);
239+
graph::eq_labeled(&g, &g2)?;
240240

241241
let g = LabeledVecGraph::<_>::from_arcs(arcs);
242242
let coo = ArcListGraph::new_labeled(g.num_nodes(), arcs.iter().copied());
243243
let g2 = LabeledVecGraph::<_>::from_lender(coo.iter());
244244

245-
assert_eq!(g, g2);
245+
graph::eq_labeled(&g, &g2)?;
246246

247247
Ok(())
248248
}

webgraph/tests/test_btree_graph.rs

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,14 @@ fn test_remove() {
1616

1717
#[cfg(feature = "serde")]
1818
#[test]
19-
fn test_serde() {
19+
fn test_serde() -> anyhow::Result<()> {
20+
use webgraph::traits::graph;
21+
2022
let arcs = [(0, 1, 1), (0, 2, 2), (1, 2, 3)];
2123

2224
let g = LabeledBTreeGraph::<usize>::from_arcs(arcs);
2325
let res = serde_json::to_string(&g).unwrap();
2426
let p: LabeledBTreeGraph<usize> = serde_json::from_str(&res).unwrap();
25-
assert_eq!(g, p);
27+
graph::eq_labeled(&g, &p)?;
28+
Ok(())
2629
}

webgraph/tests/test_transpose.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ use dsi_bitstream::traits::{BitRead, BitWrite};
1111
use webgraph::graphs::vec_graph::LabeledVecGraph;
1212
use webgraph::prelude::{transpose, transpose_labeled};
1313
use webgraph::traits::labels::SequentialLabeling;
14-
use webgraph::traits::{BitDeserializer, BitSerializer};
14+
use webgraph::traits::{graph, BitDeserializer, BitSerializer};
1515
use webgraph::utils::sort_pairs::{BitReader, BitWriter};
1616

1717
#[test]
@@ -26,7 +26,7 @@ fn test_transpose() -> anyhow::Result<()> {
2626
let trans = transpose(g2, 3)?;
2727
let g3 = VecGraph::from_lender(&trans);
2828

29-
assert_eq!(g, g3);
29+
graph::eq(&g, &g3)?;
3030
Ok(())
3131
}
3232

@@ -95,6 +95,6 @@ fn test_transpose_labeled() -> anyhow::Result<()> {
9595

9696
let g4 = LabeledVecGraph::from_lender(g.iter());
9797

98-
assert_eq!(g3, g4);
98+
graph::eq_labeled(&g3, &g4)?;
9999
Ok(())
100100
}

webgraph/tests/test_vec_graph.rs

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,20 +5,23 @@
55
* SPDX-License-Identifier: Apache-2.0 OR LGPL-2.1-or-later
66
*/
77

8+
use webgraph::traits::graph;
9+
810
#[cfg(feature = "serde")]
911
#[test]
10-
fn test_serde() {
12+
fn test_serde() -> anyhow::Result<()> {
1113
use webgraph::graphs::vec_graph::LabeledVecGraph;
1214
let arcs = [(0, 1, 1), (0, 2, 2), (1, 2, 3)];
1315

1416
let g = LabeledVecGraph::<usize>::from_arcs(arcs);
1517
let res = serde_json::to_string(&g).unwrap();
1618
let p: LabeledVecGraph<usize> = serde_json::from_str(&res).unwrap();
17-
assert_eq!(g, p);
19+
graph::eq_labeled(&g, &p)?;
20+
Ok(())
1821
}
1922

2023
#[test]
21-
fn test_epserde() {
24+
fn test_epserde() -> anyhow::Result<()> {
2225
use epserde::prelude::*;
2326
use webgraph::graphs::vec_graph::LabeledVecGraph;
2427
let arcs = [(0, 1, 1), (0, 2, 2), (1, 2, 3)];
@@ -29,5 +32,6 @@ fn test_epserde() {
2932
g.serialize(&mut file).unwrap();
3033
let data = file.into_inner();
3134
let g2 = <LabeledVecGraph<usize>>::deserialize_eps(&data).unwrap();
32-
assert_eq!(g, g2);
35+
graph::eq_labeled(&g, &g2)?;
36+
Ok(())
3337
}

0 commit comments

Comments
 (0)