Added tests for CASTable.rename

Author: Michael Erickson

swat/tests/cas/test_table.py

@@ -5174,6 +5174,38 @@ def test_partition_inputs(self):
         pcolinfo = pcolinfo.drop('ID', errors='ignore').drop('Label', errors='ignore')
         self.assertTablesEqual(colinfo[['Two', 'Model', 'One', 'MSRP']], pcolinfo)
 
+    def test_rename(self):
+        tbl = self.table
+
+        # Rename by name:
+        makeCol = tbl['Make']
+        self.assertTrue(any(col in 'Make' for col in list(tbl.columns)))
+        tbl.rename({'Make':'Manufacturer'})
+        # No column named "Make" and a column named "Manufacturer"
+        self.assertFalse(any(col in 'Make' for col in list(tbl.columns)))
+        self.assertTrue(any(col in 'Manufacturer' for col in list(tbl.columns)))
+        # Shouldn't be a "new" column, just 'Make' renamed
+        self.assertEqual(makeCol, tbl['Manufacturer'])
+
+        # Rename by function:
+        # Column Manufacturer -> Manufacturer_0
+        tbl.rename(lambda col : col + "_0")
+        for col in list(tbl.columns):
+            # Last two characters should be _0 for each col
+            self.assertEqual(col[-2:], "_0")
+
+        # Rename by name for col that doesn't exist
+        # errors='ignore'
+        originalCols = list(copy.deepcopy(tbl.columns))
+        tbl.rename({'nope':'nuh uh'})
+        self.assertFalse(any(col in 'nope' for col in list(tbl.columns)))
+        self.assertListEqual(originalCols, list(tbl.columns))
+
+        # Rename by name for col that doesn't exist
+        # errors='raise'
+        with self.assertRaises(KeyError):
+            tbl.rename(tbl.rename({'nope':'nuh uh'}, errors='raise'))
+
     def test_reset_index(self):
         tbl = self.table