type-inference

Author: ioannisPApapadopoulos

src/arrays.jl

@@ -71,7 +71,7 @@ function *(P::NDimsPlan, f::AbstractArray)
     g = copy(f)
     t = 1:ndims(g)
     for d in dims
-        perm = ntuple(k -> k == 1 ? t[d] : k == d ? t[1] : t[k], length(t))
+        perm = ntuple(k -> k == 1 ? t[d] : k == d ? t[1] : t[k], ndims(g))
         gp = permutedims(g, perm)
         g = permutedims(F*gp, inv_perm(perm))
     end
@@ -84,7 +84,7 @@ function \(P::NDimsPlan, f::AbstractArray)
     g = copy(f)
     t = 1:ndims(g)
     for d in dims
-        perm = ntuple(k -> k == 1 ? t[d] : k == d ? t[1] : t[k], length(t))
+        perm = ntuple(k -> k == 1 ? t[d] : k == d ? t[1] : t[k], ndims(g))
         gp = permutedims(g, perm)
         g = permutedims(F\gp, inv_perm(perm))
     end

test/arraystests.jl

@@ -31,6 +31,12 @@ import FastTransforms: ArrayPlan, NDimsPlan
         c = randn(20,10,20)
         @test_throws ErrorException("Different size in dims axes not yet implemented in N-dimensional transform.") NDimsPlan(ArrayPlan(plan_cheb2leg(c), c), (1,2))        
 
+        c = randn(5,20)
+        F = plan_cheb2leg(c)
+        FT = ArrayPlan(F, c)
+        P = NDimsPlan(FT, (1,))
+        @test F*c ≈ FT*c ≈ P*c
+
         c = randn(20,20,5);
         F = plan_cheb2leg(c)
         FT = ArrayPlan(F, c)
@@ -40,7 +46,6 @@ import FastTransforms: ArrayPlan, NDimsPlan
         @test size(P,1) == size(c,1)
         @test size(P,1,2) == (size(c,1), size(c,2))
 
-
         f = similar(c);
         for k in axes(f,3)
             f[:,:,k] = (F*(F*c[:,:,k])')'