diff --git a/src/forward_link.py b/src/forward_link.py index 3926abdb..1cd6aca3 100644 --- a/src/forward_link.py +++ b/src/forward_link.py @@ -42,7 +42,7 @@ def mask_values( PandasObject.mask_values = mask_values -def get_link( +def calculate_imputation_link( df: pd.DataFrame, groups: List[str] or str, match_col: str, diff --git a/tests/test_forward_link.py b/tests/test_forward_link.py index 80f35ef9..583b5485 100644 --- a/tests/test_forward_link.py +++ b/tests/test_forward_link.py @@ -2,7 +2,7 @@ import pandas as pd from pandas.testing import assert_frame_equal, assert_series_equal -from src.forward_link import get_link, mask_values +from src.forward_link import calculate_imputation_link, mask_values class TestFilters: @@ -295,7 +295,7 @@ def test_forward_link(self): ] ) - f_link = get_link( + f_link = calculate_imputation_link( self.df, ["group", "date"], "f_matched_pair", @@ -327,7 +327,7 @@ def test_backward_link(self): ] ) - b_link = get_link( + b_link = calculate_imputation_link( self.df, ["group", "date"], "b_matched_pair",