Add tests for xmap

Author: felixcremer

test/DAT/xmap.jl

@@ -0,0 +1,68 @@
+@testitem "xmap" begin
+    using YAXArrays
+    using DimensionalData
+    @testset "Loop Axis permutation" begin
+        x,y,z = X(1:4), Y(1:5), Z(1:6)
+        a1 = YAXArray((x,y,z), rand(4,5,6))
+        a2 = YAXArray((x,z,y), rand(4,6,5))
+        a3 = YAXArray((x,y), rand(4,5))
+        r = xmap(a1 ⊘ X, a2 ⊘ X, output = XOutput(dims(a1, X))) do xout, x1, x2
+            xout .= x1 .+ x2
+        end
+        @test r.data == a1.data .+ permutedims(a2.data,(1,3,2))
+        r = xmap(a2 ⊘ X, a3 ⊘ X, output=XOutput(dims(a1, X))) do xout, x1, x2
+            xout .= x1 .+ x2
+        end
+        @test r.data == a2.data .+ reshape(a3.data,(4,1,5))
+        #r = xmap(a2 ⊘ X, a3 ⊘ X, output=XOutput(dims(a1, X)),nthreads = [1]) do xout, x1, x2
+        #    xout .= x1 .+ x2
+        #end
+        @test r.data == a2.data .+ reshape(a3.data,(4,1,5))
+    end
+
+
+#=
+These should be reenabled once we decided what keyword arguments xmap gets. 
+    @testset "max cache inputs" begin
+
+        x,y,z = X(1:4), Y(1:5), Z(1:6)
+        a1 = YAXArray((x,y,z), rand(4,5,6))
+        a2 = YAXArray((x,z,y), rand(4,6,5))
+        a3 = YAXArray((x,y), rand(4,5))
+        indims = InDims("x")
+        outdims = OutDims("x")
+
+        function simple_fun(xout, x1,x2)
+            xout .= x1 .+ x2
+        end
+
+        # Float64 
+        r = xmap(simple_fun, a1 ⊘ :X, a2 ⊘ :X, output=XOutput(:X), max_cache = 6.0e8)
+        @test r.data == a1.data .+ permutedims(a2.data,(1,3,2))
+
+        # MB
+        r = mapCube(simple_fun, (a1, a2), indims=(indims, indims), outdims=outdims, max_cache = "0.5MB")
+        @test r.data == a1.data .+ permutedims(a2.data,(1,3,2))
+
+        r = mapCube(simple_fun, (a1, a2), indims=(indims, indims), outdims=outdims, max_cache = "3MB")
+        @test r.data == a1.data .+ permutedims(a2.data,(1,3,2))
+
+        r = mapCube(simple_fun, (a1, a2), indims=(indims, indims), outdims=outdims, max_cache = "10MB")
+        @test r.data == a1.data .+ permutedims(a2.data,(1,3,2))
+
+        # GB
+        r = mapCube(simple_fun, (a1, a2), indims=(indims, indims), outdims=outdims, max_cache = "0.1GB")
+        @test r.data == a1.data .+ permutedims(a2.data,(1,3,2))
+    end
+
+    @testset "Error shown in parallel" begin
+        import Zarr
+        x,y,z = X(1:4), Y(1:5), Z(1:6)
+        a1 = YAXArray((x,y,z), rand(4,5,6))
+        indims = InDims("x")
+        outdims = OutDims("x")
+        @test_throws Exception mapCube((xout, xin) -> xout .= foo(xin), a1; indims, outdims, ispar=false)
+        @test_throws CapturedException mapCube((xout, xin) -> xout .= foo(xin), a1; indims, outdims, ispar=true)
+    end
+=#
+end