assert raw also for single dtype; assert ndim == 1

Author: kernc

pandas/tests/frame/test_apply.py

@@ -236,6 +236,14 @@ def test_apply_broadcast_error(self, int_frame_const_col):
             df.apply(lambda x: Series([1, 2]), axis=1, result_type="broadcast")
 
     def test_apply_raw(self, float_frame, mixed_type_frame):
+
+        def _assert_raw(x):
+            assert isinstance(x, np.ndarray)
+            assert x.ndim == 1
+
+        float_frame.apply(_assert_raw, raw=True)
+        float_frame.apply(_assert_raw, axis=1, raw=True)
+
         result0 = float_frame.apply(np.mean, raw=True)
         result1 = float_frame.apply(np.mean, axis=1, raw=True)
 
@@ -250,10 +258,7 @@ def test_apply_raw(self, float_frame, mixed_type_frame):
         expected = float_frame * 2
         tm.assert_frame_equal(result, expected)
 
-        # Mixed dtype
-        def _assert_raw(x):
-            assert isinstance(x, np.ndarray)
-
+        # Mixed dtype (GH-32423)
         mixed_type_frame.apply(_assert_raw, raw=True)
         mixed_type_frame.apply(_assert_raw, axis=1, raw=True)