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

1"""ApproxDistinctCount feature""" 

2from abc import ABCMeta 

3from typing import Callable 

4 

5import pyspark.sql.functions as f 

6from pyspark.sql import Column 

7 

8from .feature import BaseFeature 

9 

10 

11class ApproxDistinctCount(BaseFeature, metaclass=ABCMeta): 

12 def aggregator(self) -> Callable[[Column], Column]: 

13 return self.approx_count_distinct_aggregator 

14 

15 def default_value(self) -> Column: 

16 return f.lit(0)