Rename get_link function to calculate_imputation_link

ONSdigital · May 16, 2024 · 3509145 · 3509145
1 parent d31535f
commit 3509145
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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",