refactor: consolidate some tests/examples/benchmarks
This commit is contained in:
1 parent
49be497964
commit
b8204c4ed9
83 files changed
+569
-10068
No files matched your search
@@ -52,33 +52,6 @@ async fn test_tracing_with_subscriber() {
|
||||
assert_eq!(dag.get(sum).unwrap(), 5);
|
||||
}
|
||||
|
||||
#[tokio::test(flavor = "multi_thread")]
|
||||
async fn test_tracing_with_complex_dag() {
|
||||
// Initialize tracing subscriber
|
||||
let _ = fmt()
|
||||
.with_env_filter(EnvFilter::from_default_env())
|
||||
.with_test_writer()
|
||||
.try_init();
|
||||
|
||||
let dag = DagRunner::new();
|
||||
|
||||
// Create a diamond pattern
|
||||
let source = dag.add_task(Value(10)).into();
|
||||
let left = dag.add_task(Add).depends_on((&source, &source));
|
||||
let right = dag.add_task(Multiply).depends_on((&source, &source));
|
||||
let sink = dag.add_task(Add).depends_on((&left, &right));
|
||||
|
||||
dag.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
// source = 10
|
||||
// left = 10 + 10 = 20
|
||||
// right = 10 * 10 = 100
|
||||
// sink = 20 + 100 = 120
|
||||
assert_eq!(dag.get(sink).unwrap(), 120);
|
||||
}
|
||||
|
||||
#[tokio::test(flavor = "multi_thread")]
|
||||
async fn test_tracing_inline_execution() {
|
||||
// Initialize tracing subscriber
|
||||
@@ -103,38 +76,3 @@ async fn test_tracing_inline_execution() {
|
||||
// c = 2 * 2 = 4
|
||||
assert_eq!(dag.get(c).unwrap(), 4);
|
||||
}
|
||||
|
||||
#[tokio::test(flavor = "multi_thread")]
|
||||
async fn test_tracing_multiple_layers() {
|
||||
// Initialize tracing subscriber
|
||||
let _ = fmt()
|
||||
.with_env_filter(
|
||||
EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("debug")),
|
||||
)
|
||||
.with_test_writer()
|
||||
.try_init();
|
||||
|
||||
let dag = DagRunner::new();
|
||||
|
||||
// Layer 0
|
||||
let v1 = dag.add_task(Value(1));
|
||||
let v2 = dag.add_task(Value(2)).into();
|
||||
let v3 = dag.add_task(Value(3));
|
||||
|
||||
// Layer 1
|
||||
let sum12 = dag.add_task(Add).depends_on((v1, &v2));
|
||||
let mul23 = dag.add_task(Multiply).depends_on((&v2, v3));
|
||||
|
||||
// Layer 2
|
||||
let final_sum = dag.add_task(Add).depends_on((&sum12, &mul23));
|
||||
|
||||
dag.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
// v1 = 1, v2 = 2, v3 = 3
|
||||
// sum12 = 1 + 2 = 3
|
||||
// mul23 = 2 * 3 = 6
|
||||
// final_sum = 3 + 6 = 9
|
||||
assert_eq!(dag.get(final_sum).unwrap(), 9);
|
||||
}
|
||||
Reference in new issue
Block a user