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_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')
def test_col_exists(self):
df = pd.DataFrame({'a': [1, 2, 3]})
ensure_col_exists(df, 'a')
