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

1"""MaxGrossPrice feature""" 

2import pyspark.sql.functions as f 

3from pyspark.sql import Column 

4 

5from .max_feature import Max 

6 

7 

8class MaxGrossPrice(Max): 

9 def column_expression(self) -> Column: 

10 return f.col("GrossSpend") / f.col("Quantity") 

11 

12 @property 

13 def description_subject(self) -> str: 

14 return "Maximum of (GrossSpend / Quantity)" 

15 

16 @property 

17 def commentary(self) -> str: 

18 return ( 

19 "The gross price is calculated as the gross " 

20 + "spend divided by how many were bought." 

21 )