tests

Author: vpuri3

test/func.jl

@@ -97,12 +97,34 @@ end
 
     f(du,u,p,t) = mul!(du, Diagonal(p*t), u)
 
-    op = FunctionOperator(f, u, u; p=zero(p), t=zero(t))
-
-    op = cache_operator(op, u, u)
+    L = FunctionOperator(f, u, u; p=zero(p), t=zero(t))
 
     ans = @. u * p * t
-    @test op(u,p,t) ≈ ans
-    v=copy(u); @test op(v,u,p,t) ≈ ans
+    @test L(u,p,t) ≈ ans
+    v=copy(u); @test L(v,u,p,t) ≈ ans
+
+    A = Diagonal(p * t)
+
+    u1 = rand(N, K)
+    u2 = rand(N, K)
+
+    v1 = L * u1
+    v2 = L * u2
+    @test v1 + v2 ≈ A * (u1 + u2)
+
+    v1 .= 0.0
+    v2 .= 0.0
+
+    mul!(v1, L, u1)
+    mul!(v2, L, u2)
+    @test v1 + v2 ≈ A * (u1 + u2)
+
+    v1 = rand(N, K); w1 = copy(v1)
+    v2 = rand(N, K); w2 = copy(v2)
+    a1, a2, b1, b2 = rand(4)
+
+    mul!(v1, L, u1, a1, b1)
+    mul!(v2, L, u2, a2, b2)
+    @test v1 + v2 ≈ (a1*A*u1 + b1*w1) + (a2*A*u2 + b2*w2)
 end
 #