File tree 4 files changed +8
-6
lines changed
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:
136
136
buffers .append (buf )
137
137
138
138
pandas_df = pd .DataFrame (columns )
139
- pandas_df ._buffers = buffers
139
+ pandas_df .attrs [ "_EXCHANGE_PROTOCOL_BUFFERS" ] = buffers
140
140
return pandas_df
141
141
142
142
Original file line number Diff line number Diff line change 3
3
import pandas as pd
4
4
5
5
6
- @pytest .fixture ( scope = "package" )
6
+ @pytest .fixture
7
7
def df_from_dict ():
8
8
def maker (dct , is_categorical = False ):
9
9
df = pd .DataFrame (dct )
Original file line number Diff line number Diff line change @@ -84,10 +84,12 @@ def test_dataframe(data):
84
84
indices = (0 , 2 )
85
85
names = tuple (list (data .keys ())[idx ] for idx in indices )
86
86
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 )
91
93
92
94
93
95
def test_missing_from_masked ():
You can’t perform that action at this time.
0 commit comments