From ccef63e91472e24db54f8540fcd3650d522b1e26 Mon Sep 17 00:00:00 2001
From: Connor Fitzgerald <connorwadefitzgerald@gmail.com>
Date: Mon, 21 Jun 2021 15:39:33 -0400
Subject: [PATCH] Isolate hello-compute examples to own file

---
 wgpu/examples/hello-compute/main.rs  | 94 +---------------------------
 wgpu/examples/hello-compute/tests.rs | 91 +++++++++++++++++++++++++++
 2 files changed, 92 insertions(+), 93 deletions(-)
 create mode 100644 wgpu/examples/hello-compute/tests.rs

diff --git a/wgpu/examples/hello-compute/main.rs b/wgpu/examples/hello-compute/main.rs
index 5fa86897be..086509a13c 100644
--- a/wgpu/examples/hello-compute/main.rs
+++ b/wgpu/examples/hello-compute/main.rs
@@ -196,96 +196,4 @@ fn main() {
 }
 
 #[cfg(all(test, not(target_arch = "wasm32")))]
-mod tests {
-    #[path = "../../../tests/common/mod.rs"]
-    mod common;
-
-    use std::sync::Arc;
-
-    use super::*;
-    use common::{initialize_test, TestParameters};
-
-    #[test]
-    fn test_compute_1() {
-        initialize_test(TestParameters::default(), |ctx| {
-            let input = &[1, 2, 3, 4];
-
-            pollster::block_on(assert_execute_gpu(
-                &ctx.device,
-                &ctx.queue,
-                input,
-                &[0, 1, 7, 2],
-            ));
-        });
-    }
-
-    #[test]
-    fn test_compute_2() {
-        initialize_test(TestParameters::default(), |ctx| {
-            let input = &[5, 23, 10, 9];
-
-            pollster::block_on(assert_execute_gpu(
-                &ctx.device,
-                &ctx.queue,
-                input,
-                &[5, 15, 6, 19],
-            ));
-        });
-    }
-
-    #[test]
-    fn test_compute_overflow() {
-        initialize_test(TestParameters::default(), |ctx| {
-            let input = &[77031, 837799, 8400511, 63728127];
-            pollster::block_on(assert_execute_gpu(
-                &ctx.device,
-                &ctx.queue,
-                input,
-                &[350, 524, OVERFLOW, OVERFLOW],
-            ));
-        });
-    }
-
-    #[test]
-    fn test_multithreaded_compute() {
-        initialize_test(TestParameters::default(), |ctx| {
-            use std::{sync::mpsc, thread, time::Duration};
-
-            let ctx = Arc::new(ctx);
-
-            let thread_count = 8;
-
-            let (tx, rx) = mpsc::channel();
-            for _ in 0..thread_count {
-                let tx = tx.clone();
-                let ctx = Arc::clone(&ctx);
-                thread::spawn(move || {
-                    let input = &[100, 100, 100];
-                    pollster::block_on(assert_execute_gpu(
-                        &ctx.device,
-                        &ctx.queue,
-                        input,
-                        &[25, 25, 25],
-                    ));
-                    tx.send(true).unwrap();
-                });
-            }
-
-            for _ in 0..thread_count {
-                rx.recv_timeout(Duration::from_secs(10))
-                    .expect("A thread never completed.");
-            }
-        });
-    }
-
-    async fn assert_execute_gpu(
-        device: &wgpu::Device,
-        queue: &wgpu::Queue,
-        input: &[u32],
-        expected: &[u32],
-    ) {
-        if let Some(produced) = execute_gpu_inner(device, queue, input).await {
-            assert_eq!(produced, expected);
-        }
-    }
-}
+mod tests;
diff --git a/wgpu/examples/hello-compute/tests.rs b/wgpu/examples/hello-compute/tests.rs
new file mode 100644
index 0000000000..f0945c5a80
--- /dev/null
+++ b/wgpu/examples/hello-compute/tests.rs
@@ -0,0 +1,91 @@
+#[path = "../../tests/common/mod.rs"]
+mod common;
+
+use std::sync::Arc;
+
+use super::*;
+use common::{initialize_test, TestParameters};
+
+#[test]
+fn test_compute_1() {
+    initialize_test(TestParameters::default(), |ctx| {
+        let input = &[1, 2, 3, 4];
+
+        pollster::block_on(assert_execute_gpu(
+            &ctx.device,
+            &ctx.queue,
+            input,
+            &[0, 1, 7, 2],
+        ));
+    });
+}
+
+#[test]
+fn test_compute_2() {
+    initialize_test(TestParameters::default(), |ctx| {
+        let input = &[5, 23, 10, 9];
+
+        pollster::block_on(assert_execute_gpu(
+            &ctx.device,
+            &ctx.queue,
+            input,
+            &[5, 15, 6, 19],
+        ));
+    });
+}
+
+#[test]
+fn test_compute_overflow() {
+    initialize_test(TestParameters::default(), |ctx| {
+        let input = &[77031, 837799, 8400511, 63728127];
+        pollster::block_on(assert_execute_gpu(
+            &ctx.device,
+            &ctx.queue,
+            input,
+            &[350, 524, OVERFLOW, OVERFLOW],
+        ));
+    });
+}
+
+#[test]
+fn test_multithreaded_compute() {
+    initialize_test(TestParameters::default(), |ctx| {
+        use std::{sync::mpsc, thread, time::Duration};
+
+        let ctx = Arc::new(ctx);
+
+        let thread_count = 8;
+
+        let (tx, rx) = mpsc::channel();
+        for _ in 0..thread_count {
+            let tx = tx.clone();
+            let ctx = Arc::clone(&ctx);
+            thread::spawn(move || {
+                let input = &[100, 100, 100];
+                pollster::block_on(assert_execute_gpu(
+                    &ctx.device,
+                    &ctx.queue,
+                    input,
+                    &[25, 25, 25],
+                ));
+                tx.send(true).unwrap();
+            });
+        }
+
+        for _ in 0..thread_count {
+            rx.recv_timeout(Duration::from_secs(10))
+                .expect("A thread never completed.");
+        }
+    });
+}
+
+async fn assert_execute_gpu(
+    device: &wgpu::Device,
+    queue: &wgpu::Queue,
+    input: &[u32],
+    expected: &[u32],
+) {
+    if let Some(produced) = execute_gpu_inner(device, queue, input).await {
+        assert_eq!(produced, expected);
+    }
+}