Skip to content

Commit ab4cbb7

Browse files
Add keepdims=True, bool and complex dtypes
1 parent 76eafac commit ab4cbb7

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

tests/test_mathematical.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1057,14 +1057,10 @@ def test_sum_empty_out(dtype):
10571057
(1, 10),
10581058
],
10591059
)
1060-
@pytest.mark.parametrize(
1061-
"dtype_in", get_all_dtypes(no_complex=True, no_bool=True)
1062-
)
1063-
@pytest.mark.parametrize(
1064-
"dtype_out", get_all_dtypes(no_complex=True, no_bool=True)
1065-
)
1060+
@pytest.mark.parametrize("dtype_in", get_all_dtypes())
1061+
@pytest.mark.parametrize("dtype_out", get_all_dtypes())
10661062
@pytest.mark.parametrize("transpose", [True, False])
1067-
@pytest.mark.parametrize("keepdims", [False])
1063+
@pytest.mark.parametrize("keepdims", [True, False])
10681064
def test_sum(shape, dtype_in, dtype_out, transpose, keepdims):
10691065
size = numpy.prod(shape)
10701066
a_np = numpy.arange(size).astype(dtype_in).reshape(shape)

0 commit comments

Comments
 (0)