Skip to content

Commit b3d1ee0

Browse files
Added more test cases for CSR
1 parent 0ec1442 commit b3d1ee0

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

tests/sparse/test_csr.pyi

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,12 +27,38 @@ assert_type(csr_array((2,)), csr_array[np.float64, tuple[int]])
2727
assert_type(csr_array((2, 3)), csr_array[np.float64, tuple[int, int]])
2828
assert_type(csr_array((2,), dtype=np.bool_), csr_array[np.bool_, tuple[int]])
2929
assert_type(csr_array((2, 3), dtype=np.bool_), csr_array[np.bool_, tuple[int, int]])
30+
assert_type(csr_array((2,), dtype=bool), csr_array[np.bool_, tuple[int]])
31+
assert_type(csr_array((2, 3), dtype=bool), csr_array[np.bool_, tuple[int, int]])
32+
assert_type(csr_array((2,), dtype="bool"), csr_array[np.bool_, tuple[int]])
33+
assert_type(csr_array((2, 3), dtype="bool"), csr_array[np.bool_, tuple[int, int]])
3034
assert_type(csr_array((2,), dtype=np.int64), csr_array[np.int64, tuple[int]])
3135
assert_type(csr_array((2, 3), dtype=np.int64), csr_array[np.int64, tuple[int, int]])
36+
assert_type(csr_array((2,), dtype=int), csr_array[np.int64, tuple[int]])
37+
assert_type(csr_array((2, 3), dtype=int), csr_array[np.int64, tuple[int, int]])
38+
assert_type(csr_array((2,), dtype="int"), csr_array[np.int64, tuple[int]])
39+
assert_type(csr_array((2, 3), dtype="int"), csr_array[np.int64, tuple[int, int]])
3240
assert_type(csr_array((2,), dtype=np.float64), csr_array[np.float64, tuple[int]])
3341
assert_type(csr_array((2, 3), dtype=np.float64), csr_array[np.float64, tuple[int, int]])
42+
assert_type(csr_array((2,), dtype=float), csr_array[np.float64, tuple[int]])
43+
assert_type(csr_array((2, 3), dtype=float), csr_array[np.float64, tuple[int, int]])
44+
assert_type(csr_array((2,), dtype="float"), csr_array[np.float64, tuple[int]])
45+
assert_type(csr_array((2, 3), dtype="float"), csr_array[np.float64, tuple[int, int]])
3446
assert_type(csr_array((2,), dtype=np.complex128), csr_array[np.complex128, tuple[int]])
3547
assert_type(csr_array((2, 3), dtype=np.complex128), csr_array[np.complex128, tuple[int, int]])
48+
assert_type(csr_array((2,), dtype=complex), csr_array[np.complex128, tuple[int]])
49+
assert_type(csr_array((2, 3), dtype=complex), csr_array[np.complex128, tuple[int, int]])
50+
assert_type(csr_array((2,), dtype="complex"), csr_array[np.complex128, tuple[int]])
51+
assert_type(csr_array((2, 3), dtype="complex"), csr_array[np.complex128, tuple[int, int]])
52+
assert_type(csr_array((2,), None, None), csr_array[np.float64, tuple[int]])
53+
assert_type(csr_array((2, 3), None, None), csr_array[np.float64, tuple[int, int]])
54+
assert_type(csr_array((2,), None, np.bool_), csr_array[np.bool_, tuple[int]])
55+
assert_type(csr_array((2, 3), None, np.bool_), csr_array[np.bool_, tuple[int, int]])
56+
assert_type(csr_array((2,), None, np.int64), csr_array[np.int64, tuple[int]])
57+
assert_type(csr_array((2, 3), None, np.int64), csr_array[np.int64, tuple[int, int]])
58+
assert_type(csr_array((2,), None, np.float64), csr_array[np.float64, tuple[int]])
59+
assert_type(csr_array((2, 3), None, np.float64), csr_array[np.float64, tuple[int, int]])
60+
assert_type(csr_array((2,), None, np.complex128), csr_array[np.complex128, tuple[int]])
61+
assert_type(csr_array((2, 3), None, np.complex128), csr_array[np.complex128, tuple[int, int]])
3662
assert_type(csr_matrix((2, 3)), csr_matrix[np.float64])
3763

3864
assert_type(csr_array(seq_bool), csr_array[np.bool_, tuple[int]])

0 commit comments

Comments
 (0)