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

1"""BasketCount feature""" 

2import pyspark.sql.functions as f 

3from pyspark.sql import Column 

4 

5from .distinctcount_feature import DistinctCount 

6 

7 

8class BasketCount(DistinctCount): 

9 def column_expression(self) -> Column: 

10 return f.col("Basket") 

11 

12 @property 

13 def description_subject(self) -> str: 

14 return "Distinct count of Baskets" 

15 

16 @property 

17 def commentary(self) -> str: 

18 return ( 

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

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

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

22 + "existed in that collection of transactions." 

23 )