fix helper functions for testing

OpenCOMPES · Oct 9, 2023 · bdcbacc · bdcbacc
1 parent edcab8b
commit bdcbacc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_preprocessing.py b/tests/test_preprocessing.py
@@ -16,11 +16,11 @@
 df = pd.DataFrame(np.random.randn(N_PTS, len(cols)), columns=cols)
 
 def square(df,col):
-    df[f'col_squared'] = df[col] ** 2
+    df[f'{col}_squared'] = df[col] ** 2
     return df
 
 def add(df,col,value):
-    df[f'col_added'] = df[col] + value
+    df[f'{col}_added'] = df[col] + value
     return df
 
 def test_pp_decorator():