From c640bbac95eaea99880e01269a198d90d7f7cca5 Mon Sep 17 00:00:00 2001 From: Sebastian Berg Date: Wed, 1 Jun 2022 19:10:29 -0700 Subject: [PATCH] TST: Enable inteer array assignment test and add flat test --- numpy/core/tests/test_casting_floatingpoint_errors.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/numpy/core/tests/test_casting_floatingpoint_errors.py b/numpy/core/tests/test_casting_floatingpoint_errors.py index 7248e7a0899b..87ebc194bebc 100644 --- a/numpy/core/tests/test_casting_floatingpoint_errors.py +++ b/numpy/core/tests/test_casting_floatingpoint_errors.py @@ -124,7 +124,7 @@ def integer_array_assignment(): arr[[0, 1]] = values - #yield integer_array_assignment + yield integer_array_assignment def integer_array_assignment_with_subspace(): arr = np.empty((5, 3), dtype=dtype) @@ -134,6 +134,12 @@ def integer_array_assignment_with_subspace(): yield integer_array_assignment_with_subspace + def flat_assignment(): + arr = np.empty((3,), dtype=dtype) + values = np.array([value, value, value]) + arr.flat[:] = values + + yield flat_assignment @pytest.mark.parametrize(["value", "dtype"], values_and_dtypes()) @pytest.mark.filterwarnings("ignore::numpy.ComplexWarning")