fixup! Add at-par macro

JuliaParallel · Oct 23, 2020 · 64b8a3b · 64b8a3b
1 parent d64f22d
commit 64b8a3b
Showing 1 changed file with 19 additions and 15 deletions.
diff --git a/test/thunk.jl b/test/thunk.jl
@@ -1,19 +1,23 @@
 @testset "@par" begin
-    x = 2
-    a = @par x + x
-    @test a isa Dagger.Thunk
-    b = @par sum([x,1,2])
-    c = @par a * b
-    @test collect(c) == 20
-
-    c = @par begin
+    @testset "per-call" begin
         x = 2
-        a = x + x
-        b = sum([x,1,2])
-        c = a * b
+        a = @par x + x
+        @test a isa Dagger.Thunk
+        b = @par sum([x,1,2])
+        c = @par a * b
+        @test collect(c) == 20
+    end
+
+    @testset "block" begin
+        c = @par begin
+            x = 2
+            a = x + x
+            b = sum([x,1,2])
+            c = a * b
+        end
+        @test x isa Int
+        @test a isa Dagger.Thunk
+        @test c isa Dagger.Thunk
+        @test collect(c) == 20
     end
-    @test x isa Int
-    @test a isa Dagger.Thunk
-    @test c isa Dagger.Thunk
-    @test collect(c) == 20
 end