Comparator implementation for the GE table expectation tutorial.

comparator.py

from enum import Enum, auto
from statistics import mean


class SupportedComparisonEnum(Enum):
    """Enum class with the currently supported comparison type."""

    ABSOLUTE = auto()
    MEAN = auto()

    def __call__(self, *args, **kwargs):
        if self.name == "ABSOLUTE":
            return all(args[0] >= i * args[2] / 100 for i in args[1])
        elif self.name == "MEAN":
            return args[0] >= mean(args[1]) * args[2] / 100
        else:
            raise NotImplementedError("Comparison key is not supported.")