Skip to content

Commit 5f17cf8

Browse files
committed
test with real and complex data types.
1 parent bae9724 commit 5f17cf8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/test_Scipy.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,10 @@ def test_solver(a_matrix, n_rhs, solver):
5858

5959
npt.assert_allclose(x, b, atol=tol)
6060

61-
def test_iterative_solver_linear_op():
61+
@pytest.mark.parametrize('dtype', [np.float64, np.complex128])
62+
def test_iterative_solver_linear_op(dtype):
6263
n = 10
63-
A = aslinearoperator(sp.eye(n))
64+
A = aslinearoperator(sp.eye(n).astype(dtype))
6465

6566
Ainv = SolverCG(A)
6667

0 commit comments

Comments
 (0)