Update tests, examples, benches

benches/realistic/config_broadcast.rs

@@ -2,6 +2,7 @@
 
 use criterion::{BenchmarkId, Criterion};
 use dagx::{task_fn, DagRunner};
+use futures::FutureExt;
 use std::collections::HashMap;
 use std::hint::black_box;
 
@@ -51,11 +52,9 @@ pub fn bench_config_broadcast(c: &mut Criterion) {
                             .depends_on(&config);
                         }
 
-                        dag.run(|fut| {
-                            tokio::spawn(fut);
-                        })
-                        .await
-                        .unwrap();
+                        dag.run(|fut| tokio::spawn(fut).map(Result::unwrap))
+                            .await
+                            .unwrap();
                     })
                 });
             },

benches/realistic/etl.rs

@@ -2,6 +2,7 @@
 
 use criterion::{BenchmarkId, Criterion};
 use dagx::{task_fn, DagRunner};
+use futures::FutureExt;
 
 pub fn bench_etl_pipeline(c: &mut Criterion) {
     let rt = tokio::runtime::Runtime::new().unwrap();
@@ -90,9 +91,7 @@ pub fn bench_etl_pipeline(c: &mut Criterion) {
                         ))
                         .depends_on((&validate, &scores, &names));
 
-                        dag.run(|fut| {
-                            tokio::spawn(fut);
-                        })
+                dag.run(|fut| tokio::spawn(fut).map(Result::unwrap))
                         .await
                         .unwrap();
                     })

benches/realistic/memory_efficiency.rs

@@ -2,6 +2,7 @@
 
 use criterion::{BenchmarkId, Criterion};
 use dagx::{task_fn, DagRunner};
+use futures::FutureExt;
 use std::hint::black_box;
 
 pub fn bench_memory_efficiency(c: &mut Criterion) {
@@ -38,11 +39,9 @@ pub fn bench_memory_efficiency(c: &mut Criterion) {
                             .depends_on(&source);
                         }
 
-                        dag.run(|fut| {
-                            tokio::spawn(fut);
-                        })
-                        .await
-                        .unwrap();
+                        dag.run(|fut| tokio::spawn(fut).map(Result::unwrap))
+                            .await
+                            .unwrap();
                     })
                 });
             },

benches/realistic/mixed.rs

@@ -2,6 +2,7 @@
 
 use criterion::Criterion;
 use dagx::{task_fn, DagRunner};
+use futures::FutureExt;
 
 pub fn bench_mixed_patterns(c: &mut Criterion) {
     let rt = tokio::runtime::Runtime::new().unwrap();
@@ -50,11 +51,9 @@ pub fn bench_mixed_patterns(c: &mut Criterion) {
                     .depends_on(summary);
                 }
 
-                dag.run(|fut| {
-                    tokio::spawn(fut);
-                })
-                .await
-                .unwrap();
+                dag.run(|fut| tokio::spawn(fut).map(Result::unwrap))
+                    .await
+                    .unwrap();
             })
         });
     });