Source code for prodigy.analysis.solver.mock_solver

import logging
from typing import Tuple, Optional, List, Dict

import sympy

from prodigy.analysis.solver.solver import Solver
from prodigy.distribution import Distribution
from prodigy.util.logger import log_setup



[docs]
class MockTrueSolver(Solver):
    logger = log_setup("MockTrueSolver", logging.DEBUG)


[docs]
    def solve(self, f: Distribution, g: Distribution) -> Tuple[Optional[bool], List[Dict[sympy.Expr, sympy.Expr]]]:
        return True, []





[docs]
class MockFalseSolver(Solver):
    logger = log_setup("MockFalseSolver", logging.DEBUG)


[docs]
    def solve(self, f: Distribution, g: Distribution) -> Tuple[Optional[bool], List[Dict[sympy.Expr, sympy.Expr]]]:
        return False, []





[docs]
class MockNoneSolver(Solver):
    logger = log_setup("MockNoneSolver", logging.DEBUG)


[docs]
    def solve(self, f: Distribution, g: Distribution) -> Tuple[Optional[bool], List[Dict[sympy.Expr, sympy.Expr]]]:
        return None, []