TST: Enable inteer array assignment test and add flat test

numpy · Jun 2, 2022 · c640bba · c640bba
1 parent 68e7803
commit c640bba
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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")