diff options
| author | Nathan Goldbaum <nathan.goldbaum@gmail.com> | 2023-05-03 09:33:42 -0600 |
|---|---|---|
| committer | Nathan Goldbaum <nathan.goldbaum@gmail.com> | 2023-05-03 09:33:42 -0600 |
| commit | be08e88c544ce6e1df093f649a3d9afbce9c4171 (patch) | |
| tree | ba62a3afea5aaa8600b8c2ed5c2455795aff6328 /numpy/core | |
| parent | 905d37ee1c9ebdb0ce37e1bb190a64480b4b1315 (diff) | |
| download | numpy-be08e88c544ce6e1df093f649a3d9afbce9c4171.tar.gz | |
TST: test object dtype in test_repeat
Diffstat (limited to 'numpy/core')
| -rw-r--r-- | numpy/core/tests/test_multiarray.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/numpy/core/tests/test_multiarray.py b/numpy/core/tests/test_multiarray.py index 984047c87..196c2dc13 100644 --- a/numpy/core/tests/test_multiarray.py +++ b/numpy/core/tests/test_multiarray.py @@ -1934,8 +1934,9 @@ class TestMethods: assert_array_equal(a2.prod(axis=-1), np.array([24, 1890, 600], ctype)) - def test_repeat(self): - m = np.array([1, 2, 3, 4, 5, 6]) + @pytest.mark.parametrize('dtype', [None, object]) + def test_repeat(self, dtype): + m = np.array([1, 2, 3, 4, 5, 6], dtype=dtype) m_rect = m.reshape((2, 3)) A = m.repeat([1, 3, 2, 1, 1, 2]) |