Coverage for packages/sql-smith/src/sql_smith/capability/has_offset.py: 90%

1from sql_smith.functions import literal 

2 

3 

4class HasOffsetMixin: 

5 def offset(self, new_offset: int | None = None): 

6 self._offset = new_offset 

7 return self 

8 

9 @property 

10 def has_offset(self): 

11 """Return True when the query has an offset.""" 

12 return self._offset is not None 

13 

14 def _apply_offset(self, query: "ExpressionInterface") -> "ExpressionInterface": 

15 return ( 

16 query.append("OFFSET {}", literal(self._offset)) if self._offset else query 

17 )