Use a real (probing-generated) scorer in benchmarks

[rust-lightning] / lightning / src / routing / gossip.rs

diff --git a/lightning/src/routing/gossip.rs b/lightning/src/routing/gossip.rs
index 045772486ba7aa150636a1e7b5420a606e87ce64..be1c31ff8633047c9a3f93919a4407b0025be086 100644 (file)
--- a/lightning/src/routing/gossip.rs
+++ b/lightning/src/routing/gossip.rs
@@ -3613,7 +3613,7 @@ pub mod benches {
 
        pub fn read_network_graph(bench: &mut Criterion) {
                let logger = crate::util::test_utils::TestLogger::new();
-               let mut d = crate::routing::router::bench_utils::get_route_file().unwrap();
+               let (mut d, _) = crate::routing::router::bench_utils::get_graph_scorer_file().unwrap();
                let mut v = Vec::new();
                d.read_to_end(&mut v).unwrap();
                bench.bench_function("read_network_graph", |b| b.iter(||
@@ -3623,7 +3623,7 @@ pub mod benches {
 
        pub fn write_network_graph(bench: &mut Criterion) {
                let logger = crate::util::test_utils::TestLogger::new();
-               let mut d = crate::routing::router::bench_utils::get_route_file().unwrap();
+               let (mut d, _) = crate::routing::router::bench_utils::get_graph_scorer_file().unwrap();
                let net_graph = NetworkGraph::read(&mut d, &logger).unwrap();
                bench.bench_function("write_network_graph", |b| b.iter(||
                        black_box(&net_graph).encode()