Skip to content

Commit 22360ce

Browse files
committed
linalg: type-tests for det
1 parent 493fd46 commit 22360ce

File tree

1 file changed

+22
-2
lines changed

1 file changed

+22
-2
lines changed

tests/linalg/test__basic.pyi

Lines changed: 22 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import numpy as np
66
import optype.numpy as onp
77
import optype.numpy.compat as npc
88

9-
from scipy.linalg import inv, solve, solve_banded, solve_circulant, solve_toeplitz, solve_triangular
9+
from scipy.linalg import det, inv, solve, solve_banded, solve_circulant, solve_toeplitz, solve_triangular
1010

1111
b1_nd: onp.ArrayND[np.bool_]
1212

@@ -369,7 +369,27 @@ assert_type(inv(c128_nd), onp.ArrayND[np.complex128])
369369
assert_type(inv(c160_nd), onp.ArrayND[np.complex128])
370370

371371
###
372-
# TODO(jorenham): det
372+
# det
373+
374+
assert_type(det(f32_2d), np.float64)
375+
assert_type(det(f64_2d), np.float64)
376+
assert_type(det(c64_2d), np.complex128)
377+
assert_type(det(c128_2d), np.complex128)
378+
379+
assert_type(det(py_b_2d), np.float64)
380+
assert_type(det(py_i_2d), np.float64)
381+
assert_type(det(py_f_2d), np.float64)
382+
assert_type(det(py_c_2d), np.complex128)
383+
384+
assert_type(det(f32_3d), onp.Array1D[np.float64])
385+
assert_type(det(f64_3d), onp.Array1D[np.float64])
386+
assert_type(det(c64_3d), onp.Array1D[np.complex128])
387+
assert_type(det(c128_3d), onp.Array1D[np.complex128])
388+
389+
assert_type(det(py_b_3d), onp.Array1D[np.float64])
390+
assert_type(det(py_i_3d), onp.Array1D[np.float64])
391+
assert_type(det(py_f_3d), onp.Array1D[np.float64])
392+
assert_type(det(py_c_3d), onp.Array1D[np.complex128])
373393

374394
###
375395
# TODO(jorenham): lstsq

0 commit comments

Comments
 (0)