def test_is_column_subset(self): # columns_labels_missing contains column names that aren't in df1, so this should raise an exception self.assertRaises(MissingColumns, is_column_subset, self.column_labels_missing, self.df1.columns.tolist()) # column_labels contains column names that are all in df1, so is_column_subset should not raise an exception! try: is_column_subset(self.column_labels, self.df1.columns.tolist()) except MissingColumns: self.fail("is_column_subset() raised MissingColumns unexpectedly!")
def transform(self, df, **transform_params): # sanity checks is_dataframe(df) is_column_subset( self.col_labels, df.columns.tolist() ) # extract columns by label return df[self.col_labels]