def test_basic(self, simple_df): """ Ensure the functions are applied to the dataframe """ df = simple_df td = np.timedelta64(1, "s") funcs = {"time": lambda x: x + td, "latitude": lambda x: x + 1} out = upd.apply_funcs_to_columns(simple_df, funcs) # a dataframe copy should have been returned assert out is not df # very the functions were applied assert (out["time"] == df["time"] + td).all() assert (out["latitude"] == df["latitude"] + 1).all()
def test_inplace(self, simple_df): """ Inplace should be, well, inplace. """ funcs = {"latitude": lambda x: x + 1} out = upd.apply_funcs_to_columns(simple_df, funcs, inplace=True) assert out is simple_df
def test_skips_missing_column(self, simple_df): """ A function should not be called on a non-existent column. """ funcs = {"bob": lambda x: x / 0} out = upd.apply_funcs_to_columns(simple_df, funcs) assert "bob" not in out.columns