Coverage for addmo/s2_data_tuning/feature_constructor.py: 100%

1import pandas as pd 

2 

3"""If the name of function is "create_" + "variable suffix", the methods are dynamically used  

4through the getattr() function in the DataTunerByConfig class.""" 

5 

6def create_lag(var: pd.Series, lag: int): 

7 """ 

8 Creates a lagged version of the input series. 

9 """ 

10 series = var.shift(lag) 

11 series.name = f"{var.name}___lag{lag}" 

12 return series 

13 

14def create_diff(var: pd.Series): 

15 """ 

16 Creates a differenced version of the input series. 

17 """ 

18 series = var.diff() 

19 series.name = f"{var.name}___diff" 

20 return series 

21 

22def create_squared(var: pd.Series): 

23 """ 

24 Creates a squared version of the input series. 

25 """ 

26 series = var.pow(2) 

27 series.name = f"{var.name}___squared" 

28 return series