Coverage for jstark / mealkit / cuisines.py: 100%

1"""Cuisines feature""" 

2 

3import pyspark.sql.functions as f 

4from pyspark.sql import Column 

5 

6from jstark.features.collect_set_feature import CollectSet 

7 

8 

9class Cuisines(CollectSet): 

10 def column_expression(self) -> Column: 

11 return f.col("Cuisine") 

12 

13 @property 

14 def description_subject(self) -> str: 

15 return "Set of Cuisines" 

16 

17 @property 

18 def commentary(self) -> str: 

19 return ( 

20 "The set of cuisines. Typically the dataframe supplied " 

21 + "to this feature will have many recipes for the same cuisine, " 

22 + "this feature allows you to determine the set of cuisines " 

23 + "that have been ordered." 

24 )