Merge pull request #20 from ChrisRackauckas/a_mul_b

fix A_mul_B!
JuliaLinearAlgebra · Aug 22, 2017 · 0df1421 · 0df1421
2 parents c59afab + 082f709
commit 0df1421
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/strang.jl b/src/strang.jl
@@ -11,7 +11,7 @@ function getindex{T}(S::Strang{T}, i, j)
     abs(i - j) == 1 && return -1
     0
 end
-getindex{T}(S::Strang{T}, I...) = getindex(S,ind2sub(size(S),I)...)
+getindex{T}(S::Strang{T}, I...) = getindex(S,ind2sub(size(S),I...)...)
 size(S::Strang, r::Int) = r==1 || r==2 ? S.n : throw(ArgumentError("Invalid dimension $r"))
 size(S::Strang) = S.n, S.n
 full{T}(S::Strang{T}) = full(strang(T, S.n))

diff --git a/test/strang.jl b/test/strang.jl
@@ -10,6 +10,12 @@ for i in 1:n, j in 1:n
     abs(i-j)>1 && @test Z[i,j] == 0
 end
 
+A = Strang(10)
+u = ones(10)
+v = similar(u)
+A_mul_B!(v,A,u)
+@test v == [1.0;zeros(8);1.0]
+
 #Matvec product
 b = randn(n)
 @test_approx_eq Z*b full(Z)*b