Add test case for _InverseLinearOperator

probabilistic-numerics · Nov 8, 2022 · d5fffce · d5fffce
1 parent 3f2043c
commit d5fffce
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/test_linops/test_linops_cases/linear_operator_cases.py b/tests/test_linops/test_linops_cases/linear_operator_cases.py
@@ -91,3 +91,20 @@ def case_sparse_matrix_singular(
     )
 
     return pn.linops.Matrix(matrix), matrix.toarray()
+
+
+@pytest.mark.parametrize("rng", [np.random.default_rng(422)])
+def case_inverse(
+    rng: np.random.Generator,
+) -> Tuple[pn.linops.LinearOperator, np.ndarray]:
+    N = 100
+
+    v = rng.uniform(0.2, 0.5, N)
+
+    linop = pn.linops.LambdaLinearOperator(
+        shape=(N, N),
+        dtype=np.double,
+        matmul=lambda x: 2.0 * x + v[:, None] @ (v[None, :] @ x),
+    )
+
+    return linop, linop.todense()