@@ -95,7 +95,15 @@ def test_df_get_chunks(size, n_chunks, df_from_dict):
95
95
df = df_from_dict ({"x" : list (range (size ))})
96
96
dfX = df .__dataframe__ ()
97
97
chunks = list (dfX .get_chunks (n_chunks ))
98
- assert len (chunks ) == n_chunks
98
+ """
99
+ Pandas do not plan to use more chunks than 1.
100
+ But if chunking is supported in your implementation,
101
+ then it is the place to use the following check
102
+
103
+ ``assert len(chunks) == n_chunks``
104
+
105
+ """
106
+ assert len (chunks ) == 1
99
107
assert sum (chunk .num_rows () for chunk in chunks ) == size
100
108
101
109
@@ -104,7 +112,15 @@ def test_column_get_chunks(size, n_chunks, df_from_dict):
104
112
df = df_from_dict ({"x" : list (range (size ))})
105
113
dfX = df .__dataframe__ ()
106
114
chunks = list (dfX .get_column (0 ).get_chunks (n_chunks ))
107
- assert len (chunks ) == n_chunks
115
+ """
116
+ Pandas do not plan to use more chunks than 1.
117
+ But if chunking is supported in your implementation,
118
+ then it is the place to use the following check
119
+
120
+ ``assert len(chunks) == n_chunks``
121
+
122
+ """
123
+ assert len (chunks ) == 1
108
124
assert sum (chunk .size for chunk in chunks ) == size
109
125
110
126
@@ -129,7 +145,7 @@ def test_buffer(df_from_dict):
129
145
130
146
assert dataBuf .bufsize > 0
131
147
assert dataBuf .ptr != 0
132
- device , _ = dataBuf .__dlpack_device__
148
+ device = dataBuf .__dlpack_device__
133
149
134
150
assert dataDtype [0 ] == 0
135
151
0 commit comments