File tree Expand file tree Collapse file tree 4 files changed +8
-6
lines changed Expand file tree Collapse file tree 4 files changed +8
-6
lines changed Original file line number Diff line number Diff line change @@ -136,7 +136,7 @@ def protocol_df_chunk_to_pandas(df: DataFrameXchg) -> pd.DataFrame:
136136 buffers .append (buf )
137137
138138 pandas_df = pd .DataFrame (columns )
139- pandas_df ._buffers = buffers
139+ pandas_df .attrs [ "_EXCHANGE_PROTOCOL_BUFFERS" ] = buffers
140140 return pandas_df
141141
142142
Original file line number Diff line number Diff line change 33import pandas as pd
44
55
6- @pytest .fixture ( scope = "package" )
6+ @pytest .fixture
77def df_from_dict ():
88 def maker (dct , is_categorical = False ):
99 df = pd .DataFrame (dct )
Original file line number Diff line number Diff line change @@ -84,10 +84,12 @@ def test_dataframe(data):
8484 indices = (0 , 2 )
8585 names = tuple (list (data .keys ())[idx ] for idx in indices )
8686
87- tm .assert_frame_equal (
88- from_dataframe (df2 .select_columns (indices )),
89- from_dataframe (df2 .select_columns_by_name (names )),
90- )
87+ result = from_dataframe (df2 .select_columns (indices ))
88+ expected = from_dataframe (df2 .select_columns_by_name (names ))
89+ tm .assert_frame_equal (result , expected )
90+
91+ assert isinstance (result .attrs ["_EXCHANGE_PROTOCOL_BUFFERS" ], list )
92+ assert isinstance (expected .attrs ["_EXCHANGE_PROTOCOL_BUFFERS" ], list )
9193
9294
9395def test_missing_from_masked ():
You can’t perform that action at this time.
0 commit comments