def test_col_does_not_exist(self): df = pd.DataFrame({'a': [1, 2, 3]}) with self.assertRaises(ValueError): ensure_col_exists(df, 'b', 'my_df')
def test_col_exists(self): df = pd.DataFrame({'a': [1, 2, 3]}) ensure_col_exists(df, 'a')