pmap tests

Author: nmheim

Project.toml

@@ -28,6 +28,7 @@ StructArrays = "0.6"
 julia = "1.10"
 
 [extras]
+Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
 FiniteDifferences = "26cc04aa-876d-5657-8c51-4c34ba976000"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -36,4 +37,4 @@ Symbolics = "0c5d862f-8b57-4792-8d23-62f2024744c7"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["ForwardDiff", "LinearAlgebra", "Random", "Symbolics", "Test", "FiniteDifferences"]
+test = ["Distributed", "FiniteDifferences", "ForwardDiff", "LinearAlgebra", "Random", "Symbolics", "Test"]

test/gradcheck.jl

@@ -10,6 +10,7 @@
 using Test
 using ChainRulesCore
 using Diffractor
+using Distributed: pmap
 using FiniteDifferences
 using LinearAlgebra
 
@@ -283,23 +284,24 @@ end
         @test jacobicheck(xs -> map(x -> x^2, xs), rand(2,3))
 
         # MethodError: no method matching copy(::Nothing)
-        @test_broken jacobicheck((xss...) -> sum(map((xs...) -> sqrt(sum(xs.^2)), xss...)), [rand(5) for _ in 1:6]...)
+        @test_broken jacobicheck((xss...) -> map((xs...) -> sqrt(sum(xs.^2)), xss...), [rand(5) for _ in 1:6]...)
 
         # MethodError: no method matching copy(::Nothing)
         @test_broken gradcheck(y -> map(x -> x*y, 1:5), 3)
         @test_broken gradient(v -> sum([x for x in v]), [1.1,2.2,3.3]) == ([1, 1, 1],)
     end
 
-    @test_skip @testset "bascis, pmap" begin
-        @test jacobicheck(xs -> sum(pmap(x -> x^2, xs)), rand(2,3))
-        @test jacobicheck((xss...) -> sum(pmap((xs...) -> sqrt(sum(xs.^2)), xss...)), [rand(5) for _ in 1:6]...)
+    @testset "bascis, pmap" begin
+        # BoundsError: attempt to access 0-element Core.Compiler.UnitRange{Int64} at index [0]
+        @test_broken jacobicheck(xs -> pmap(x -> x^2, xs), rand(2,3))
+        @test_broken jacobicheck((xss...) -> pmap((xs...) -> sqrt(sum(xs.^2)), xss...), [rand(5) for _ in 1:6]...)
 
         function foo(y)
             bar = (x) -> x*y
             sum(pmap(bar, 1:5))
         end
-        @test gradtest(foo, 3)
-        @test gradient(v -> sum([x for x in v]), [1.1,2.2,3.3]) == ([1, 1, 1],)
+        # BoundsError: attempt to access 0-element Core.Compiler.UnitRange{Int64} at index [0]
+        @test_broken jacobicheck(y -> pmap(x -> x*y, 1:5), 3)
     end
 
     @testset "Tuple adjoint" begin
@@ -402,4 +404,3 @@ end
 #     end
 #     @test gradcheck(bar, [5.0])
 # end
-