Skip to content

Commit 365524f

Browse files
author
Michael Erickson
committed
Completed tests for CASTable.rename
1 parent 72a5949 commit 365524f

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

swat/tests/cas/test_table.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5176,30 +5176,38 @@ def test_partition_inputs(self):
51765176

51775177
def test_rename(self):
51785178
tbl = self.table
5179+
df = self.get_cars_df()
51795180

51805181
# Rename by name:
51815182
makeCol = tbl['Make']
51825183
self.assertTrue(any(col in 'Make' for col in list(tbl.columns)))
51835184
tbl.rename({'Make': 'Manufacturer'})
5185+
df.rename(columns={'Make': 'Manufacturer'}, inplace=True)
51845186
# No column named "Make" and a column named "Manufacturer"
51855187
self.assertFalse(any(col in 'Make' for col in list(tbl.columns)))
51865188
self.assertTrue(any(col in 'Manufacturer' for col in list(tbl.columns)))
51875189
# Shouldn't be a "new" column, just 'Make' renamed
51885190
self.assertEqual(makeCol, tbl['Manufacturer'])
5191+
# Shoud be same as pandas
5192+
self.assertTablesEqual(tbl, df)
51895193

51905194
# Rename by function:
51915195
# Column Manufacturer -> Manufacturer_0
51925196
tbl.rename(lambda col: col + "_0")
5197+
df.rename(columns=lambda col: col + "_0", inplace=True)
51935198
for col in list(tbl.columns):
51945199
# Last two characters should be _0 for each col
51955200
self.assertEqual(col[-2:], "_0")
5201+
self.assertTablesEqual(tbl, df)
51965202

51975203
# Rename by name for col that doesn't exist
51985204
# errors='ignore'
51995205
originalCols = list(copy.deepcopy(tbl.columns))
52005206
tbl.rename({'nope': 'nuh uh'})
5207+
df.rename(columns={'nope': 'nuh uh'}, inplace=True)
52015208
self.assertFalse(any(col in 'nope' for col in list(tbl.columns)))
52025209
self.assertListEqual(originalCols, list(tbl.columns))
5210+
self.assertTablesEqual(tbl, df)
52035211

52045212
# Rename by name for col that doesn't exist
52055213
# errors='raise'

0 commit comments

Comments
 (0)