1
1
from datetime import datetime
2
- import inspect
3
2
4
3
import numpy as np
5
4
import pytest
@@ -137,34 +136,6 @@ def test_dti_set_index_reindex(self):
137
136
138
137
# Renaming
139
138
140
- def test_reindex_api_equivalence (self ):
141
- # equivalence of the labels/axis and index/columns API's
142
- df = DataFrame (
143
- [[1 , 2 , 3 ], [3 , 4 , 5 ], [5 , 6 , 7 ]],
144
- index = ["a" , "b" , "c" ],
145
- columns = ["d" , "e" , "f" ],
146
- )
147
-
148
- res1 = df .reindex (["b" , "a" ])
149
- res2 = df .reindex (index = ["b" , "a" ])
150
- res3 = df .reindex (labels = ["b" , "a" ])
151
- res4 = df .reindex (labels = ["b" , "a" ], axis = 0 )
152
- res5 = df .reindex (["b" , "a" ], axis = 0 )
153
- for res in [res2 , res3 , res4 , res5 ]:
154
- tm .assert_frame_equal (res1 , res )
155
-
156
- res1 = df .reindex (columns = ["e" , "d" ])
157
- res2 = df .reindex (["e" , "d" ], axis = 1 )
158
- res3 = df .reindex (labels = ["e" , "d" ], axis = 1 )
159
- for res in [res2 , res3 ]:
160
- tm .assert_frame_equal (res1 , res )
161
-
162
- res1 = df .reindex (index = ["b" , "a" ], columns = ["e" , "d" ])
163
- res2 = df .reindex (columns = ["e" , "d" ], index = ["b" , "a" ])
164
- res3 = df .reindex (labels = ["b" , "a" ], axis = 0 ).reindex (labels = ["e" , "d" ], axis = 1 )
165
- for res in [res2 , res3 ]:
166
- tm .assert_frame_equal (res1 , res )
167
-
168
139
def test_assign_columns (self , float_frame ):
169
140
float_frame ["hi" ] = "there"
170
141
@@ -173,38 +144,6 @@ def test_assign_columns(self, float_frame):
173
144
tm .assert_series_equal (float_frame ["C" ], df ["baz" ], check_names = False )
174
145
tm .assert_series_equal (float_frame ["hi" ], df ["foo2" ], check_names = False )
175
146
176
- def test_rename_signature (self ):
177
- sig = inspect .signature (DataFrame .rename )
178
- parameters = set (sig .parameters )
179
- assert parameters == {
180
- "self" ,
181
- "mapper" ,
182
- "index" ,
183
- "columns" ,
184
- "axis" ,
185
- "inplace" ,
186
- "copy" ,
187
- "level" ,
188
- "errors" ,
189
- }
190
-
191
- def test_reindex_signature (self ):
192
- sig = inspect .signature (DataFrame .reindex )
193
- parameters = set (sig .parameters )
194
- assert parameters == {
195
- "self" ,
196
- "labels" ,
197
- "index" ,
198
- "columns" ,
199
- "axis" ,
200
- "limit" ,
201
- "copy" ,
202
- "level" ,
203
- "method" ,
204
- "fill_value" ,
205
- "tolerance" ,
206
- }
207
-
208
147
209
148
class TestIntervalIndex :
210
149
def test_setitem (self ):
@@ -256,21 +195,6 @@ def test_set_reset_index(self):
256
195
257
196
258
197
class TestCategoricalIndex :
259
- def test_set_index_preserve_categorical_dtype (self ):
260
- # GH13743, GH13854
261
- df = DataFrame (
262
- {
263
- "A" : [1 , 2 , 1 , 1 , 2 ],
264
- "B" : [10 , 16 , 22 , 28 , 34 ],
265
- "C1" : Categorical (list ("abaab" ), categories = list ("bac" ), ordered = False ),
266
- "C2" : Categorical (list ("abaab" ), categories = list ("bac" ), ordered = True ),
267
- }
268
- )
269
- for cols in ["C1" , "C2" , ["A" , "C1" ], ["A" , "C2" ], ["C1" , "C2" ]]:
270
- result = df .set_index (cols ).reset_index ()
271
- result = result .reindex (columns = df .columns )
272
- tm .assert_frame_equal (result , df )
273
-
274
198
@pytest .mark .parametrize (
275
199
"codes" , ([[0 , 0 , 1 , 1 ], [0 , 1 , 0 , 1 ]], [[0 , 0 , - 1 , 1 ], [0 , 1 , 0 , 1 ]])
276
200
)
0 commit comments