numpy ravel with dataframe test GH#26247

Signed-off-by: Liang Yan <[email protected]>

Author: liang3zy22

pandas/tests/arrays/test_array.py

@@ -429,3 +429,22 @@ def test_array_to_numpy_na():
     result = arr.to_numpy(na_value=True, dtype=bool)
     expected = np.array([True, True])
     tm.assert_numpy_array_equal(result, expected)
+
+
+def test_array_ravel_from_df():
+    # GH26247
+    x = np.random.randn(10, 3)
+
+    result = np.ravel([pd.DataFrame(batch.reshape(1, 3)) for batch in x])
+    expected = np.ravel(x)
+    tm.assert_numpy_array_equal(result, expected)
+
+    result = np.ravel(pd.DataFrame(x[0].reshape(1, 3), columns=["x1", "x2", "x3"]))
+    expected = x[0]
+    tm.assert_numpy_array_equal(result, expected)
+
+    result = np.ravel(
+        [pd.DataFrame(batch.reshape(1, 3), columns=["x1", "x2", "x3"]) for batch in x]
+    )
+    expected = np.ravel(x)
+    tm.assert_numpy_array_equal(result, expected)