diff --git a/python/pyarrow/tests/test_pandas.py b/python/pyarrow/tests/test_pandas.py index 0734d0581f7e0..60c483164a0f9 100644 --- a/python/pyarrow/tests/test_pandas.py +++ b/python/pyarrow/tests/test_pandas.py @@ -2828,8 +2828,12 @@ def test_partial_schema(self): def test_table_batch_empty_dataframe(self): df = pd.DataFrame({}) - _check_pandas_roundtrip(df) - _check_pandas_roundtrip(df, as_batch=True) + _check_pandas_roundtrip(df, preserve_index=None) + _check_pandas_roundtrip(df, preserve_index=None, as_batch=True) + + expected = pd.DataFrame(columns=pd.Index([])) + _check_pandas_roundtrip(df, expected, preserve_index=False) + _check_pandas_roundtrip(df, expected, preserve_index=False, as_batch=True) df2 = pd.DataFrame({}, index=[0, 1, 2]) _check_pandas_roundtrip(df2, preserve_index=True)