Coverage for jstark / grocery / basket_count.py: 100%

1"""BasketCount feature""" 

2 

3import pyspark.sql.functions as f 

4from pyspark.sql import Column 

5 

6from jstark.features.distinctcount_feature import DistinctCount 

7 

8 

9class BasketCount(DistinctCount): 

10 def column_expression(self) -> Column: 

11 return f.col("Basket") 

12 

13 @property 

14 def description_subject(self) -> str: 

15 return "Distinct count of Baskets" 

16 

17 @property 

18 def commentary(self) -> str: 

19 return ( 

20 "The number of baskets. Typically the dataframe supplied " 

21 + "to this feature will have many transactions for the same basket, " 

22 + "this feature allows you to determine how many shopping baskets " 

23 + "existed in that collection of transactions." 

24 )