Coverage for jstark/features/max_gross_spend.py: 100%

9 statements  

« prev     ^ index     » next       coverage.py v7.4.3, created at 2024-02-25 20:09 +0000

1"""MaxGrossSpend feature""" 

2import pyspark.sql.functions as f 

3from pyspark.sql import Column 

4 

5from .max_feature import Max 

6 

7 

8class MaxGrossSpend(Max): 

9 def column_expression(self) -> Column: 

10 return f.col("GrossSpend") 

11 

12 @property 

13 def description_subject(self) -> str: 

14 return "Maximum GrossSpend value"