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

1"""Quantity feature""" 

2 

3import pyspark.sql.functions as f 

4from pyspark.sql import Column 

5 

6from jstark.features.sum_feature import Sum 

7 

8 

9class Quantity(Sum): 

10 def column_expression(self) -> Column: 

11 return f.col("Quantity") 

12 

13 @property 

14 def description_subject(self) -> str: 

15 return "Sum of Quantity" 

16 

17 def default_value(self) -> Column: 

18 return f.lit(0)