@@ -129,16 +129,18 @@ def test_pass_readonly_array():
129129
130130def test_nonunit_stride_from_python ():
131131 from pybind11_tests import (
132- double_row , double_col , double_mat_cm , double_mat_rm ,
132+ double_row , double_col , double_complex , double_mat_cm , double_mat_rm ,
133133 double_threec , double_threer )
134134
135135 counting_mat = np .arange (9.0 , dtype = np .float32 ).reshape ((3 , 3 ))
136136 second_row = counting_mat [1 , :]
137137 second_col = counting_mat [:, 1 ]
138138 np .testing .assert_array_equal (double_row (second_row ), 2.0 * second_row )
139139 np .testing .assert_array_equal (double_col (second_row ), 2.0 * second_row )
140+ np .testing .assert_array_equal (double_complex (second_row ), 2.0 * second_row )
140141 np .testing .assert_array_equal (double_row (second_col ), 2.0 * second_col )
141142 np .testing .assert_array_equal (double_col (second_col ), 2.0 * second_col )
143+ np .testing .assert_array_equal (double_complex (second_col ), 2.0 * second_col )
142144
143145 counting_3d = np .arange (27.0 , dtype = np .float32 ).reshape ((3 , 3 , 3 ))
144146 slices = [counting_3d [0 , :, :], counting_3d [:, 0 , :], counting_3d [:, :, 0 ]]
@@ -564,14 +566,17 @@ def test_special_matrix_objects():
564566
565567
566568def test_dense_signature (doc ):
567- from pybind11_tests import double_col , double_row , double_mat_rm
569+ from pybind11_tests import double_col , double_row , double_complex , double_mat_rm
568570
569571 assert doc (double_col ) == """
570572 double_col(arg0: numpy.ndarray[float32[m, 1]]) -> numpy.ndarray[float32[m, 1]]
571573 """
572574 assert doc (double_row ) == """
573575 double_row(arg0: numpy.ndarray[float32[1, n]]) -> numpy.ndarray[float32[1, n]]
574576 """
577+ assert doc (double_complex ) == """
578+ double_complex(arg0: numpy.ndarray[complex64[m, 1]]) -> numpy.ndarray[complex64[m, 1]]
579+ """
575580 assert doc (double_mat_rm ) == """
576581 double_mat_rm(arg0: numpy.ndarray[float32[m, n]]) -> numpy.ndarray[float32[m, n]]
577582 """
0 commit comments