@@ -5176,30 +5176,38 @@ def test_partition_inputs(self):
5176
5176
5177
5177
def test_rename (self ):
5178
5178
tbl = self .table
5179
+ df = self .get_cars_df ()
5179
5180
5180
5181
# Rename by name:
5181
5182
makeCol = tbl ['Make' ]
5182
5183
self .assertTrue (any (col in 'Make' for col in list (tbl .columns )))
5183
5184
tbl .rename ({'Make' : 'Manufacturer' })
5185
+ df .rename (columns = {'Make' : 'Manufacturer' }, inplace = True )
5184
5186
# No column named "Make" and a column named "Manufacturer"
5185
5187
self .assertFalse (any (col in 'Make' for col in list (tbl .columns )))
5186
5188
self .assertTrue (any (col in 'Manufacturer' for col in list (tbl .columns )))
5187
5189
# Shouldn't be a "new" column, just 'Make' renamed
5188
5190
self .assertEqual (makeCol , tbl ['Manufacturer' ])
5191
+ # Shoud be same as pandas
5192
+ self .assertTablesEqual (tbl , df )
5189
5193
5190
5194
# Rename by function:
5191
5195
# Column Manufacturer -> Manufacturer_0
5192
5196
tbl .rename (lambda col : col + "_0" )
5197
+ df .rename (columns = lambda col : col + "_0" , inplace = True )
5193
5198
for col in list (tbl .columns ):
5194
5199
# Last two characters should be _0 for each col
5195
5200
self .assertEqual (col [- 2 :], "_0" )
5201
+ self .assertTablesEqual (tbl , df )
5196
5202
5197
5203
# Rename by name for col that doesn't exist
5198
5204
# errors='ignore'
5199
5205
originalCols = list (copy .deepcopy (tbl .columns ))
5200
5206
tbl .rename ({'nope' : 'nuh uh' })
5207
+ df .rename (columns = {'nope' : 'nuh uh' }, inplace = True )
5201
5208
self .assertFalse (any (col in 'nope' for col in list (tbl .columns )))
5202
5209
self .assertListEqual (originalCols , list (tbl .columns ))
5210
+ self .assertTablesEqual (tbl , df )
5203
5211
5204
5212
# Rename by name for col that doesn't exist
5205
5213
# errors='raise'
0 commit comments