Coverage for packages/sql-smith/src/sql_smith/partial/criteria.py: 100%

1from sql_smith.interfaces import CriteriaInterface 

2 

3 

4class Criteria(CriteriaInterface): 

5 def __init__(self, expression: "ExpressionInterface"): 

6 self._expression = expression 

7 

8 def and_(self, right: "CriteriaInterface") -> "CriteriaInterface": 

9 return Criteria(self._expression.append("AND {}", right)) 

10 

11 def or_(self, right: "CriteriaInterface") -> "CriteriaInterface": 

12 return Criteria(self._expression.append("OR {}", right)) 

13 

14 def sql(self, engine: "EngineInterface") -> str: 

15 return self._expression.sql(engine) 

16 

17 def params(self, engine: "EngineInterface") -> tuple: 

18 return self._expression.params(engine)