fix addvector

Author: vpuri3

src/matrix.jl

@@ -236,19 +236,26 @@ function AffineOperator(A::Union{AbstractMatrix,AbstractSciMLOperator},
     AffineOperator(A, B, b, cache, update_func)
 end
 
+"""
+    L = AddVector(b[; update_func])
+    L(u) = u + b
+"""
 function AddVector(b::AbstractVecOrMat; update_func=DEFAULT_UPDATE_FUNC)
     N  = size(b, 1)
-    Z  = NullOperator{N}()
     Id = IdentityOperator{N}()
 
-    AffineOperator(Id, B, b; update_func=update_func)
+    AffineOperator(Id, Id, b; update_func=update_func)
 end
 
+"""
+    L = AddVector(B, b[; update_func])
+    L(u) = u + B*b
+"""
 function AddVector(B, b::AbstractVecOrMat; update_func=DEFAULT_UPDATE_FUNC)
     N = size(B, 1)
-    Z = NullOperator{N}()
+    Id = IdentityOperator{N}()
 
-    AffineOperator(Z, B, b; update_func=update_func)
+    AffineOperator(Id, B, b; update_func=update_func)
 end
 
 getops(L::AffineOperator) = (L.A, L.B, L.b)

test/matrix.jl

@@ -97,6 +97,22 @@ end
     @test L \ u ≈ D \ (u - B * b)
     v=rand(N,K); @test ldiv!(v, L, u) ≈ D \ (u-B*b)
     v=copy(u); @test ldiv!(L, u) ≈ D \ (v-B*b)
+
+    L = AddVector(b)
+    @test L * u ≈ u + b
+    @test L \ u ≈ u - b
+    v=rand(N,K); @test mul!(v, L, u) ≈  u + b
+    v=rand(N,K); w=copy(v); @test mul!(v, L, u, α, β) ≈ α*(u + b) + β*w
+    v=rand(N,K); @test ldiv!(v, L, u) ≈  u - b
+    v=copy(u); @test ldiv!(L, u) ≈  v - b
+
+    L = AddVector(MatrixOperator(B), b)
+    @test L * u ≈ u + B * b
+    @test L \ u ≈ u - B * b
+    v=rand(N,K); @test mul!(v, L, u) ≈  u + B * b
+    v=rand(N,K); w=copy(v); @test mul!(v, L, u, α, β) ≈ α*(u + B * b) + β*w
+    v=rand(N,K); @test ldiv!(v, L, u) ≈  u - B * b
+    v=copy(u); @test ldiv!(L, u) ≈  v - B * b
 end
 
 @testset "AffineOperator update test" begin